Assignment for Data Scientist's Toolbox course