Shift most JS to functions.php