added module2 test and refactored to support multiple tests