Add GetModelWells function