Minor UI changes

/cc @amitbharadwaj @josemontesdeoca