FIX UI ISSUES ON SMALLER SCREENS