modify mega-menu code