Add group edit logic