Added comments to problem 5