Improve the demo.