Update build_with_pub_key.py