for question 2 of Data Science Toolkit class project