Sửa lại phần model