Made small UI changes