changed ui a bit