Step 2: Add  logic to handle zoned clusters