Working on styling with flexbox