This is a data science assignment