Fix docstrings and minor formatting fixes