added comments to prob 1