added tests - tests work as planned