add two more tests for classes