Add some ui tweaks