Updated code for mutal infor analysis