add `display: flex` support