update the demo code in readme

update the demo code in readme