Moved step_size into the state struct