Update DiversityModeling.py