STY: Fixing docstrings