Removed test from the module. Added all tests in the test class