Data Scientist Toolbox course project 1