added set_temp and set_timestep stages