changed to flexbox