Re-add main method