Data Science ToolBox class assignment