whoops, forgot to change the main method back