Created for project 1 for Data Scientist's Toolbox course on Coursera