Updated README and docstrings