Shitty python script for Data Structures and Algorithms Level 4 challenge.