Button styling and UI tweaks