Update README and docstring