Improve css for small screen