data science hw 1: problem 1 and 2