Updates to the README and code comments