@media (max-width: 520px) {
    .input-container {
        display: flex;
        align-items: center;
    }
}