Update script to generate keys.