Wordpress - add div wrapper around sub-menu

sub-menu-wrap

/* EXTEND SUBNAV
******************************************/

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

<?php wp_nav_menu(array(
	'walker' => new submenu_wrap()
)); ?>

Removed the hyphen from the class name and this works well for me in Wordpress 5.4.1. Thank you!

Really helpful - works great.

Any ideas on how I can give a different class to each level of sub, sub-sub-menu, etc. ?

Classes can be added directly in the WP Edit Menu to each menu item

This is true, and a good point but I was looking for an automated method.

Thanks for the ideas!

Please how to change ul and li to div