For the data scientist toolbox course project