     }
     
     // The cell that is returned must be retrieved from a call to -dequeueReusableCellWithReuseIdentifier:forIndexPath:
-    func collectionView(collectionView: UICollectionView, cellForItemAtIndexPath indexPath: NSIndexPath) -> UICollectionViewCell {
+    func collectionView(collectionView: AATCollectionView, cellForItemAtIndexPath indexPath: NSIndexPath) -> UICollectionViewCell {
         let cell = collectionView.dequeueReusableCellWithReuseIdentifier(cardCellReuseID, forIndexPath: indexPath)
-        configureCell(cell, forIndexPath: indexPath)
+        configureCell(cell, forIndexPath: indexPath, accordingToCardModel: collectionView.cardModel)
         return cell
     }
     

    
    
    // MARK: == Private helper methods ==
    func configureCell(cell:UICollectionViewCell, forIndexPath indexPath:NSIndexPath) {
        let imageView = UIImageView(frame: cell.contentView.bounds)
        imageView.contentMode = .ScaleAspectFit
        imageView.image = UIImage(named: "transparent-black-circle-medium")

    }
    
    //ViewController Ends here
}

// MARK: - === AATCollectionView class ===
class AATCollectionView : UICollectionView {
    var cardModel:CardID = CardID.Card1
}


