Made minor UI changes