changed to timestep return