Thanks! I have updated the code on github to include your procedure for calculating the IA2 probabilities shown in Table 7 column 3 in Pfizer's protocol spec.