Update mutual information calculation