for Data Scientist's Toolbox course project