Separate script from functions.php