Tidies up UI for mobile