Docstring improvements