Added more models for data updating