finish Prob 2 part 1