assignment 1 for data science