changed UI a bit