Update some docstrings for clarity