Update README.md

minor updates for running the code.