add comments

add comments that explains the calculation of the energy.