Update demo for v0.2.1