Clean up UI a little bit.