Update docstring
Update docstrings