Fixes, UI tweaks