diff --git a/kit/functions.php b/kit/functions.php
index e977a7e..b8e3534 100644
--- a/kit/functions.php
+++ b/kit/functions.php
@@ -31,6 +31,7 @@ add_filter( 'nav_menu_link_attributes', 'add_menu_link_class', 1, 3 );
 
 class kit_custom_main_menu extends Walker_Nav_Menu {
 
+  # Menu start, equivalent to <ul>
   function start_lvl( &$output, $depth = 0, $args = array() ) {
 
       // depth dependent classes
@@ -45,6 +46,9 @@ class kit_custom_main_menu extends Walker_Nav_Menu {
       // build html
       if ($display_depth == 1) {
           $output .= "\n" . $indent . '
+            <div class="nav-first-level-opener">
+              <span>Untermenu öffnen / schließen</span>
+            </div>
             <section class="nav-second-level mega-flyout">
               <div class="content-wrap">
                 <div class="container container-3-cols">
@@ -75,63 +79,60 @@ class kit_custom_main_menu extends Walker_Nav_Menu {
       }
   }
 
-    function start_el(&$output, $item, $depth = 0, $args=array(), $id = 0) {
-    	$object = $item->object;
-    	$type = $item->type;
-    	$title = $item->title;
-    	$description = $item->description;
-    	$permalink = $item->url;
+  # Menu element, equivalent to <li>
+  function start_el(&$output, $item, $depth = 0, $args=array(), $id = 0) {
+  	$object = $item->object;
+  	$type = $item->type;
+  	$title = $item->title;
+  	$description = $item->description;
+  	$permalink = $item->url;
 
-      $theme_template_path = get_template_directory_uri();
+    $theme_template_path = get_template_directory_uri();
 
-      if ($depth == 1) {
-        $output .= '
-        <div class="column">
-          <div class="nav-second-level-group">
-            <a class="nav-second-level-group-headline" href="' . $permalink . '">
-              <div class="icon"></div>
-              <span>
-            ';
-        $output .= $title;
-        $output .= '
-              </span>
-            </a>
-            <div class="nav-second-level-opener">
-              <span>Untermenu öffnen / schließen</span>
-            </div>';
-      } elseif ($depth == 2) {
-        $output .= "<li class='" .  implode(" ", $item->classes) . "'>";
-    	  $output .= '
-                      <a href="' . $permalink . '">';
-        $output .= $title;
-      	$output .= '  </a>';
-      } else {
-        $output .= "<li class='" .  implode(" ", $item->classes) . "'>";
-        $output .= '<div class="nav-first-level-opener">
-                      <span>Untermenu öffnen / schließen</span>
-                    </div>
-                    ';
-        $output .= '<a href="' . $permalink . '">';
-        $output .= $title;
-        $output .= '</a>';
-      }
+    if ($depth == 1) {
+      $output .= '
+      <div class="column">
+        <div class="nav-second-level-group">
+          <a class="nav-second-level-group-headline" href="' . $permalink . '">
+            <div class="icon"></div>
+            <span>
+          ';
+      $output .= $title;
+      $output .= '
+            </span>
+          </a>
+          <div class="nav-second-level-opener">
+            <span>Untermenu öffnen / schließen</span>
+          </div>';
+    } elseif ($depth == 2) {
+      $output .= "<li class='" .  implode(" ", $item->classes) . "'>";
+  	  $output .= '
+                    <a href="' . $permalink . '">';
+      $output .= $title;
+    	$output .= '  </a>';
+    } else {
+      $output .= "<li class='" .  implode(" ", $item->classes) . "'>";
+      $output .= '<a href="' . $permalink . '">';
+      $output .= $title;
+      $output .= '</a>';
     }
+  }
 
-    function end_el(&$output, $item, $depth = 0, $args=array(), $id = 0) {
-    	$object = $item->object;
-    	$type = $item->type;
-    	$title = $item->title;
-    	$description = $item->description;
-    	$permalink = $item->url;
+  function end_el(&$output, $item, $depth = 0, $args=array(), $id = 0) {
+  	$object = $item->object;
+  	$type = $item->type;
+  	$title = $item->title;
+  	$description = $item->description;
+  	$permalink = $item->url;
 
-      if ($depth == 1) {
-        $output .= '
-                  </div>
-                </div>';
-      } else {
-      	$output .= '</li>';
-      }
+    if ($depth == 1) {
+      $output .= '
+                </div>
+              </div>';
+    } else {
+    	$output .= '</li>';
     }
+  }
 
 }