Updated functions.php