Update README.md

Update the README to reflect the code improvements made.