Minor changes in UI