Updated README file and code