Added the demo code