Make the flex display work