Add flexbox and style