update demo code