Assignment two of the data scientist toolkit