Update template-functions.php