Simplify UpdateModel function