Enhance docstrings all around