Adding the demo code