Add jQuery box behaviour (functions.php)