Minor changes to code and ReadME