Improve demo2