Improve layout using flex

I like it.   flexbox ftw!