update code in readme