Maybe use `P` to be consistent with the notation in the paper.