more tweaking the flex layouts