updated the demo controller