add gene choice probs

erm, maybe some other stuff?