Data Scientist's Toolbox's course project, step 2