https://cipdpro.co.uk/program-assignment-python-4