Update README.md

Modified code to show output in comments