Updating how the demo is run