Update README.md with new code.

Fixed small bug.