This is for question 2 of The Data Scientist’s Toolbox