minor code fixes and readme udpate