Fixed up UI a bit.