made some more changes to ui