Assignment for Data Scientist Toolbox course