Made minor UI edits