This mainly goes back to a simple top down flow on the headers
Instead of using floats and fixed margins

Use margin over flex grow

Decouple button and h2