Updating diplomacy code