cleaned up UI a little bit