Modify add key script