Update test cases, add get_defined_classes