Update README and fix comment