Few changes in UI design