Added comments to problem4