added tests to main