Add diversity info to MI extract