Added test for main