updated ui a bit