fixed the UI for mobile