Impove docstrings