Updated README and docstring