Added tests for the Class struct