UI fixes

Update UI