Update README to help people run the code locally