Fixing docstring