Adding missing main method