Forgot to change last condition to actually accept a different timestep number.