Added version 0.2: energy equation in ODE solver