clean up ui a bit