Cleaning up docstring