This is the question 2 of final project or Data Scientist class