clean up mutual_information.py after Michael's refactor