added missing UI changes