added test to class