Update test demo for 0.2