Update README.md

Fixed a mistake in the code visualization