made minor changes to ui