Update README.md

Added the most relevant code to the readme