Minor changes to the demo.