https://cipdpro.co.uk/program-assignment-data-structures-python-2