Fix a docstring and formatting