Update README.md

Fix README to make it work with the current implementation