small improvement to UI