Update demo to work