增加两个model类