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