Add the main method again