UI Changes

Minor UI changes