Fix small-screen layout issues