Better AI ship construction decisions when under a treaty