Added drag force to energy