Made the flexbox responsive