made some changes to ui