update README and docstring