A little change to count correctly the number of balls implied in all the boxes