Added comments to problem 4