add tests to main test class