Don't set display:flex in js

We'll use -webkit- and -ms- prefixes in css, it will not work well in js.