The Data Scientist's Toolbox Course Project, question 2