This is for Problem 2 of the Course Project for The Data Scientist's Toolbox