Updated README and code comments