Create gen_key.py