data science assignment