diversity experiment code