Updated UI tweaks