some changes on UI