Update demo script.
Merge pull request #119 from mpark/demo

Update demo script.