Use media queries to change the column count based on screen size