Add media query for columns