Add Descriptions to the Total Theme integrated simple Megamenu

//MEGAMENU DESCRIPTION COL
function be_header_menu_desc( $item_output, $item, $depth, $args ) {
	if ( 'main_menu' == $args->theme_location && ! $depth && $item->description ) { //
		$item_output = str_replace( '</a>', '</a><span class="new-badge">' . $item->description . '</span>', $item_output );
	} else if ( 'main_menu' == $args->theme_location && $item->description ) {
		$item_output = str_replace( '</a>', '<div class="megamenu-description">' . $item->description . '</div></a>', $item_output );
	}

	return $item_output;
}
add_filter( 'walker_nav_menu_start_el', 'be_header_menu_desc', 10, 4 );