Update README.md

Updated instructions for running the code.