more tweaks on ui