Fixing docstrings