As part of the assignment for the Data Scientist's Toolbox