complete uw data science assignment1 problem 1-4