improved ui a little