introduction to data science assigment 1