Rebalance, bug fixes