change in demo code