Update demo for v0.2.2