Add a few minore UI tweaks