feat: UI tweaks