Add todo about unequal group sizes