Replace mobile with flexbox