Added support for Mutual Information