Final demo code