cleanup up UI a little