Created for the Data Scientist's Toolbox course project