Updated README and code