Fix optimize_steps docstring