[UI] Update UI a bit