Add layout optimizations for small screens