For Data Scientist's Toolbox course project