Question 2 of the Data Scientist's Toolbox course project