css improvents for flex-box