Add equality check to end proofs