Better test for classes