Minor changes in ui