.nav-menu-desktop li.menu-item-mega {
    position: relative;
}

.nav-menu-desktop li.menu-item-mega > .sub-menu {
    display: none;
    position: absolute;
    top: 100%;
    left: 0;
    width: auto;
    background: var(--white, #fff);
    border: 1px solid var(--mist-3);
    border-radius: 0;
    padding: 1.5rem;
    box-shadow: 0 8px 24px rgba(10, 22, 40, .1);
    z-index: 100;
}

.nav-menu-desktop li.menu-item-mega:hover > .sub-menu,
.nav-menu-desktop li.menu-item-mega:focus-within > .sub-menu {
    display: grid;
}

.nav-menu-desktop li.menu-item-mega.menu-item-mega-2 > .sub-menu { grid-template-columns: repeat(2, 1fr); }
.nav-menu-desktop li.menu-item-mega.menu-item-mega-3 > .sub-menu { grid-template-columns: repeat(3, 1fr); }
.nav-menu-desktop li.menu-item-mega.menu-item-mega-4 > .sub-menu { grid-template-columns: repeat(4, 1fr); }
.nav-menu-desktop li.menu-item-mega.menu-item-mega-5 > .sub-menu { grid-template-columns: repeat(5, 1fr); }

.nav-menu-desktop li.menu-item-mega > .sub-menu > li {
    padding: 0 1rem;
}

.nav-menu-desktop li.menu-item-mega:hover > .nav-submenu-toggle::before,
.nav-menu-desktop li.menu-item-mega:focus-within > .nav-submenu-toggle::before {
    transform: translateY(2px) rotate(225deg);
}

.nav-menu-desktop li.menu-item-mega > .sub-menu > li.menu-item-has-children {
    display: block;
}

.nav-menu-desktop li.menu-item-mega > .sub-menu > li > a {
    font-weight: 700;
    color: var(--ocean-2);
    margin-bottom: 0.5rem; /* Add some space below the header */
}

.nav-menu-desktop li.menu-item-mega > .sub-menu > li > .sub-menu {
    position: static;
    border: none;
    box-shadow: none;
    padding: 0;
    display: block;
}

.nav-menu-desktop li.menu-item-mega > .sub-menu > li > .sub-menu > li > a {
    font-weight: 500;
    color: var(--ink-3);
    padding: .5rem 0;
}

.nav-menu-desktop li.menu-item-mega > .sub-menu > li > .sub-menu > li > a:hover {
    color: var(--ocean-2);
}