Assignment for coursera data scientist's tool box course