working on problem set 1 problem 3