Small ui tweaks
UI tweaks