Trying to implement a timestep