Generalization through Diversity: Improving Unsupervised Environment Design