Optimizing AI for Teamwork