Fixed minor UI changes