Switching scripts away from functions.php