Add AI attack mechanics