Update README.md and fix code.