update demo code.