added tests add functionality