Small change in timestepping scheme