Added new test for Class type returned