Improve the demo_app