add flex box in media query