Add Rebalancing Tests