Finish correctnes proofs