Created as the second part of the project assignment for Data Scientist's Toolbox