Add proof for unfold identity
Shorten some proofs