Rewrite docstrings