Better flex styling