Some minor tweaks to the UI