Made changes to the demo.