add main method back in