Added model updating