UI tweaks, bugfixes