fix: add flex display to the display;