Created as an assignment for Data Scientist Toolkit course