more tweaks to the UI