added tests - tests are green