Data Science Class Assignment