Add more UI tweaks