Modify the main method