Add styling and media queries