added test for classes