Improved the calculation of the photon momentum.