UI improvements, optimize better for mobile.