Data Scientist class project