improved the demo