Adding demo code