Update demo code.