fix rk energy plot pyodeint nb