Update UI improvements