Update demo for switch