change docstring to correct the example