This is for the Data Scientist's Toolbox assignment