Task 2 in the Data Scientist's Toolbox course project