add total_energy_and_force equations