Setup demo code