closer to balanced assignment sampling