Slightly better layout using flexbox