UI tweaks & updates