data scientist toolbox assignment 1