This is Question 2 of data science project