/* Media query for smaller screens */
        @media (max-width: 768px) {
            .image-container {
                flex: 1 1 calc(50% - 10px);
            }
        }

        @media (max-width: 480px) {
            .image-container {
                flex: 1 1 100%;
            }
        }