More style tweaks
Fix flexbox inconsistencies