finishing up prob 1