Added some tests for class types