Update keygen.py