Minor changes

Updated README and made a couple of minor code changes