few changes in UI