Added better test for main