Created from assignment in the Data Scientist's Toolbox Course