Enhance docstrings