Updated UI a little