fix(demo): improve demo