add unit tests in main module