cleaned up ui a little bit