improve ui a little