Test class works