Updated README.md with code improvements