Added comments to problem 2