Add the demo code