Upgraded ai_behavior