add_filter( 'avf_main_menu_nav' , 'um_enfold_menu' , 100 , 1 ); function um_enfold_menu( $menu ){ $menu = ubermenu( 'main' , array( 'theme_location' => 'avia' , 'echo' => false ) ); return $menu; } function add_custom_script(){ ?>