Hola realxp, bienvenido a los foros
:
Pues el código es bastante sencillo y está bien... La función no importa si está en el head o en el body, pero obligatoriamente debe estar en una sección script...
Porqué no nos muestras cómo lo estás intentando... seguro que tienes algo mal.
Saludos