update demo code

update demo code