Some UI updates per feedback