https://progiez.com/the-joy-of-computing-using-python-assignment-1-answers