added tests to class