Update demo code