Use flex boxes for UI