Created for the Data Scientists' Toolbox course project