changed UI a little bit