Made UI changes