updated the the diversity notebook