updated ui a little