Update demo to v0.5.2