add flex boxes
Fix comments from Github