made some changes to the ui