Use flexbox styling