Model changes - adding category to model