Update rekey.py