Make div go under in flex container

Question

I\'m trying to implement the following responsive behaviour for a navbar.

Here is the result I want on large screens:

And on small screens (Iphone 8 an