Update styles to use media queries