def get_next_green(vehicle_counts): # Returns the road with the highest vehicle count return max(vehicle_counts, key=vehicle_counts.get)