Fixed up the demo project