Modify the demo
Merge pull request #41 from junyuecao/patch-1

Modify the demo