Updated UI for mobile