update code in README