updated test case for class