UI a little change