Data Science Toolbox Assignment 2