update for assignment 1 problem 1
update for assignment 1 problem 2