Improvements to the AI and diplomacy