data science course assignment 1