clean up UI a bit