Trying to get Mutual Information to work, still taking too long
minor changes