Updates to the demo