I added tests