Created as part of assignment for Data Scientist's Toolbox course on coursera