Update functions.php with menu functions