Making minor UI changes after some feedback.