improve demo

improve demo