added changes to model update