Added a test for main