Update README.md

Add code