Update  functions.php