private CarrierAgent getCarrierAgent(Id driverId) { if (driverAgentMap.containsKey(driverId)) { return driverAgentMap.get(driverId); } for (CarrierAgent ca : carrierAgents) { if (ca.getDriverIds().contains(driverId)) { driverAgentMap.put(driverId, ca); return ca; } } return null; }