For the 2nd task on the course project for Data Scientist's Toolbox course