Update README.md

Updated the README.md to match the latest changes in the code