some changes to the UI