cleaning up prob 2