Update protein_folding.py