Almost done with prob 1