Removed unnecessary code from demo.