         case 0:
             return 1
         case 1:
-            return 3
+            return mediaImages.count
         default:
             return 0
         }


    @IBOutlet weak var messageTextView: UITextView!
    @IBOutlet weak var mediaCollectionView: UICollectionView!

    let feedMediaAddCellID = "FeedMediaAddCell"
    let feedMediaCellID = "FeedMediaCell"

        }
    }

    override func viewDidAppear(animated: Bool) {
        super.viewDidAppear(animated)

        // touch to create (if need) for faster appear
        delay(0.2) { [weak self] in
            self?.imagePicker.hidesBarsOnTap = false
        }
    }

    // MARK: Actions

    func post(sender: UIBarButtonItem) {
