Update README.md

Add the code.