Kinetic energy calculation