Add functionality for computing mutual information