For coursera Data Scientist's Toolbox assignment