Fixing some stuff in the demo.