For the 'Data Scientist's Toolbox' course project