Réglage des hauteurs égales pour les div avec jQuery

Je veux égale hauteur divs avec jQuery.
Tous les divs peut avoir un nombre différent de contenu et des différents hauteur par défaut. Voici un échantillon de ma mise en page html:

<div class="container">
    <div class="column">This is<br />the highest<br />column</div>
    <div class="column">One line</div>
    <div class="column">Two<br />lines</div>
</div>
<div class="container">
    <div class="column">One line</div>
    <div class="column">Two<br>lines</div>
    <div class="column">One line</div>
</div>

Je suis réglage de la hauteur avec la prochaine fonction jQuery:

$(document).ready(function(){

    var highestBox = 0;
        $('.container .column').each(function(){  
                if($(this).height() > highestBox){  
                highestBox = $(this).height();  
        }
    });    
    $('.container .column').height(highestBox);

});

Cela fonctionne bien, mais pas dans mon cas car je veux que tous les "colonnes" de l'égalité qu'à l'intérieur d'un "container". Cela signifie que, dans le premier contenant toutes les cases doivent être aussi haut que le premier div, mais dans le second, ils doivent être égaux à la deuxième colonne.

Donc, la question est: comment dois-je modifier mon jQuery pour atteindre cet objectif?

Merci!