fixed up the UI a bit