Update folding.py