Added in class test cases