For class project for Data Scientists Toolbox coursera class