Fix docstrings and references