Add width for flex