finish the demo code

finish the demo code