Improved css for small screens