Fix up UI a little.