Use flexboxes where possible