improving UI

small improvement