Additional UI updates