.az-sitemap-container {
    display: flex;
    flex-wrap: wrap;
}

.az-sitemap-col {
    flex: 1 0 50%; /* On small screens, two columns */
    padding: 0 15px;
    box-sizing: border-box;
}

@media (min-width: 768px) {
    .az-sitemap-col {
        flex: 1 0 33.33%; /* On medium screens, three columns */
    }
}

@media (min-width: 992px) {
    .az-sitemap-col {
        flex: 1 0 25%; /* On large screens, four columns */
    }
}