Data Science Toolbox assignment