Updated to new model version