Clean up UI on mobile