UI tweaks, final UI