如何在swift中创建主类别/子类别数据模型？

// 定义主类别数据模型
class MainCategory {
    var name: String
    
    init(name: String) {
        self.name = name
    }
    
    func display() {
        print("Main Category: \(name)")
    }
}

// 定义子类别数据模型
class SubCategory: MainCategory {
    var subName: String
    
    init(name: String, subName: String) {
        self.subName = subName
        super.init(name: name)
    }
    
    override func display() {
        print("Main Category: \(name), Sub Category: \(subName)")
    }
}

// 创建主类别对象
let mainCategory = MainCategory(name: "Main Category")
mainCategory.display()

// 创建子类别对象
let subCategory = SubCategory(name: "Main Category", subName: "Sub Category")
subCategory.display()