For assignment 2 of the Data Scientist toolbox course project