Fix main method