Improve docstrings
Improve docstrings