     }
     
     func pickerView(pickerView: UIPickerView, numberOfRowsInComponent component: Int) -> Int {
-        return countryArray.count
+       
+        if pickerView.tag == 0 {
+        
+            return countryArray.count
+        
+        } else if pickerView.tag == 1{
+        
+            return transportationTypeArray.count
+        }
+        
+        return 1
     }
     
     func pickerView(pickerView: UIPickerView, titleForRow row: Int, forComponent component: Int) -> String? {
+       
+        if pickerView.tag == 0 {
+            return countryArray[row]
+        }
+        if pickerView.tag == 1 {
+            
+            return transportationTypeArray[row]
+        }
+        
         return countryArray[row]
     }
     
     func pickerView(pickerView: UIPickerView, didSelectRow row: Int, inComponent component: Int) {
         
+        if pickerView.tag == 0 {
+            
         let indexPath = NSIndexPath(forRow: 0, inSection: 0)
         let cell = tableView.cellForRowAtIndexPath(indexPath) as! AddTableViewCell
         cell.countryTextField.text = countryArray[row]
+            
+        } else if pickerView.tag == 1 {
+            
+            for index in 0..<rows.count {
+                
+                if let cell = allArray[index] as? TransportationTableViewCell {
+    
+                        cell.typeTextField.text = transportationTypeArray[row]
+                }
+            }
+        }
     }
     
     func pickerView(pickerView: UIPickerView, attributedTitleForRow row: Int, forComponent component: Int) -> NSAttributedString? {

        }
        
        
        
        // Picker View UI
        setUpPickerViewUI()
        
        saveButton.enabled = false
        

            
            cell.startDateTextField.inputView = startDatePicker
            cell.returnDateTextField.inputView = returnDatePicker
            
            //        startDatePicker.reloadInputViews()
            startDatePicker.addTarget(self, action: #selector(AddViewController.updateDateField(_:)), forControlEvents: .ValueChanged)
