https://cipdpro.co.uk/python-assignment-3