work on fixing timestep