some changes to UI