little ui cleanup