Docstring updat