finish Prob 2 part 2