added tests that work