Clean up UI a bit.