upgrade docs, docstring