working on the comparison; added get_prob_homo