Added probability support to genotype IO framework