Update README.md

Simplify some code snippets