rebalance code..