https://progiez.com/the-joy-of-computing-using-python-week-0-nptel-assignment-answer