Data science toolbox homework