Minor changes made to code and README