add test for main module