My data science assignment