few more changes in UI