updated return variable to modelName