added tests for classes