For question 2 of final project of "Data Scientist Toolbox"