Working but breaks with small timestep