edit README and update code