Changing docstring to reflect code