Simplify main method generation