         
         return cell
     }
-    
-    //MARK: - UIImagePickerControllerDelegate
+}
+
+extension MomentPostViewController: UIImagePickerControllerDelegate {
     func imagePickerControllerDidCancel(_ picker: UIImagePickerController) {
         // Dismiss the picker if the user canceled.
         dismiss(animated: true, completion: nil)

            fatalError("Expected a dictionary containing an image, but was provided the following: \(info)")
        }
        
        // Add uploaded image to array
        uploadedImages.append(selectedImage)
        
        // Dismiss the picker.
        dismiss(animated: true, completion: nil)
        
        momentImageCollectionView.reloadData()
    }
    
    //MARK: - UICollectionViewDelegate
    
    func collectionView(_ collectionView: UICollectionView, didSelectItemAt indexPath: IndexPath) {
        
        // response to tapping at the add button

            present(imagePickerController, animated: true, completion: nil)
        }
    }
}

extension MomentPostViewController: UITextViewDelegate {

    func textViewDidBeginEditing(_ textView: UITextView) {
        if textView.textColor == UIColor.lightGray {
            textView.text = nil
