created set of states, pRoof<1, and made argmax iterate over those instead, as per the algorithm description. Have no idea if this makes any difference as the previous implementation probably was some shortcut that I just undid. Also, as per algorithm description, argmax checks that pTilde > 1 and not pRoof > 1.