added tests class