fixed small bug in mutual_information func