Bugfixes and tweaks to the UI.