Fixed the demo
Removed erroneous main method