Switch to flexbox layout