Change main method

Change main method