Updated prove.py
Removed class proof.
Remove useless functions