https://cipdpro.co.uk/python-homework-help-2