update docstring to match code