Add media queries
Some style fix