Added implementation for model update.