Addressed review feedback for rebalance doc.