Update docstrings of the public API