.container {
    flex-flow: column wrap;
}

@media(min-width:767px) {
    .container {
        flex-flow: row wrap;
    }
}