Fix sort_groups, add tests