Update functions.php