Update README.md

Enhanced code section