update code in Readme