modified timestep function in universe