Added ability to update models