Completed problem 2 and 3 of PS2