Minor update to model