Funciones de Jquery click function hasClass removeClass addClass cambiar clase

[duralai]

Tengo la siguiente función, al pegar el código en la consola y teclear Enter, todo lo que tiene que hacer la función se ejecuta bien. Pero al pegar el código en el Head y luego abrir el documento NO PASA NADA...que tengo que hacer para que la función se ejecute al abrir el documento?

Lo que hace este código es cambiar la clase de los botones dentro de un Div con id "container" y también cambiar el estilo de une de los elementos por display "block" para mostrar el contenido.

Código: [Seleccionar]

$("#container button").click(changeButtonAtr);

function changeButtonAtr() {if ($(this).hasClass("r-collapsed")) {
    $(this).removeClass("r-collapsed");
    $(this).addClass("r-expanded");
    $(this).parent().parent().parent().next().find(".data").css("display", "block");
} else if ($(this).hasClass("r-expanded")) {
    $(this).removeClass("r-expanded");
    $(this).addClass("r-collapsed");
    $(this).parent().parent().parent().next().find(".data").css("display", "none");
}
}

[Lorenzo31]

Código: [Seleccionar]

$( document ).ready(function() {

$("#container button").on('click', function(){
    if ($(this).hasClass("r-collapsed")) {
       $(this).removeClass("r-collapsed");
       $(this).addClass("r-expanded");
       $(this).parent().parent().parent().next().find(".data").css("display", "block");
   } else if ($(this).hasClass("r-expanded")) {
       $(this).removeClass("r-expanded");
       $(this).addClass("r-collapsed");
       $(this).parent().parent().parent().next().find(".data").css("display", "none");
   }

});

});