修改 model 类