Fixing menu bug in functions.php