added UI changes