Adding main model update functionality