Update demo code
Merge pull request #27 from huei90/master

Update demo code