Slightly improve docstrings