Adding some small UI changes.