wordpress - 如何仅将 div 添加到深度 2 子菜单？

Question

我找到了这个 wordpress nav menu walker 代码，并且在添加此代码时，div 会附加到每个深度的所有子菜单。我只需要 div 用于深度 2 的子菜单。我应该改变什么以获得所需的结果。

class walker_nav_sustom_megasub extends Walker_Nav_Menu
{
    function start_lvl( &$output, $depth = 0, $args = array() ) {
    $indent = str_repeat("\t", $depth);
    $output .= "\n$indent<div class='sm-container'><ul class='sub-menu'>\n";
    }
    function end_lvl( &$output, $depth = 0, $args = array() ) {
    $indent = str_repeat("\t", $depth);
    $output .= "$indent</ul></div>\n";
    }
}