Add some minor UI tweaks