Added test for main module