Edit functions.php