Exercise for coursera data scientist's toolbox course