cleaned up ui a little