change to functions.php