Clean up docstring