one last tweak to UI