Update README.md

Changed README to reflect updated code.