Updated demo

Fixes #5