Working on rebalance()