tweaked UI a little