README updated and minor changes to the code