improve demo
improve demo