Added tweaks to ui