You need to remove the following:

.container div {
    flex: 1;
}
.search {
    flex: 1;
}

And also remove the line in bold here:

.logout {
    background-color: red;
    color: white;
    margin-left: auto;
}

Then it’ll work.