adding UI changes