Improvements to the flex items and width changes