Updated to reflect new models variable names.