Added test for classes with attributes