Re-updating diplmacy code