Improve the demo, document