fix a unit, gonna rewrite with odeint