Update README.md

Added the code.