Adding assignment 1 for data science class