Addressed review feedback on new RebalanceController.