class assignment for data science project