Function to map genotypes
change a little