Minor ui tweaks