Add step_size