Add feature to prove