added tests
Added tests