Add some changes in UI