Added time step in code