data science assignment question