Minor edit to model