UI fixes for mobile layout