Add doc comment for BalanceNode method