added tests to class controller