Add logic for grouping units