Fix remaining docstrings