Added kinetic energy calculation in force loop