diff --git a/NHBalancedFlowLayoutDemo/NHBalancedFlowLayoutDemo/Main.storyboard b/NHBalancedFlowLayoutDemo/NHBalancedFlowLayoutDemo/Main.storyboard
index d489d58..775c9b1 100644
--- a/NHBalancedFlowLayoutDemo/NHBalancedFlowLayoutDemo/Main.storyboard
+++ b/NHBalancedFlowLayoutDemo/NHBalancedFlowLayoutDemo/Main.storyboard
@@ -1,8 +1,8 @@
 <?xml version="1.0" encoding="UTF-8" standalone="no"?>
-<document type="com.apple.InterfaceBuilder3.CocoaTouch.Storyboard.XIB" version="3.0" toolsVersion="4514" systemVersion="13B3116" targetRuntime="iOS.CocoaTouch.iPad" propertyAccessControl="none" useAutolayout="YES" initialViewController="V8m-No-bba">
+<document type="com.apple.InterfaceBuilder3.CocoaTouch.Storyboard.XIB" version="3.0" toolsVersion="6245" systemVersion="14A361c" targetRuntime="iOS.CocoaTouch.iPad" propertyAccessControl="none" useAutolayout="YES" initialViewController="V8m-No-bba">
     <dependencies>
         <deployment defaultVersion="1536" identifier="iOS"/>
-        <plugIn identifier="com.apple.InterfaceBuilder.IBCocoaTouchPlugin" version="3747"/>
+        <plugIn identifier="com.apple.InterfaceBuilder.IBCocoaTouchPlugin" version="6238"/>
     </dependencies>
     <scenes>
         <!--Navigation Controller-->
@@ -29,7 +29,6 @@
                         <rect key="frame" x="0.0" y="0.0" width="1024" height="768"/>
                         <autoresizingMask key="autoresizingMask" widthSizable="YES" heightSizable="YES"/>
                         <color key="backgroundColor" white="1" alpha="1" colorSpace="calibratedWhite"/>
-                        <simulatedOrientationMetrics key="simulatedOrientationMetrics" orientation="landscapeRight"/>
                         <collectionViewFlowLayout key="collectionViewLayout" minimumLineSpacing="0.0" minimumInteritemSpacing="0.0" id="LFy-mI-zcv" customClass="NHBalancedFlowLayout">
                             <size key="itemSize" width="200" height="200"/>
                             <size key="headerReferenceSize" width="50" height="50"/>
@@ -44,19 +43,12 @@
                                     <rect key="frame" x="0.0" y="0.0" width="200" height="200"/>
                                     <autoresizingMask key="autoresizingMask"/>
                                     <subviews>
-                                        <imageView userInteractionEnabled="NO" contentMode="scaleAspectFill" horizontalHuggingPriority="251" verticalHuggingPriority="251" translatesAutoresizingMaskIntoConstraints="NO" id="pF7-Pf-FWu">
+                                        <imageView userInteractionEnabled="NO" contentMode="scaleAspectFill" horizontalHuggingPriority="251" verticalHuggingPriority="251" fixedFrame="YES" translatesAutoresizingMaskIntoConstraints="NO" id="pF7-Pf-FWu">
                                             <rect key="frame" x="0.0" y="0.0" width="200" height="200"/>
-                                            <autoresizingMask key="autoresizingMask" widthSizable="YES" heightSizable="YES"/>
                                         </imageView>
                                     </subviews>
                                     <color key="backgroundColor" white="0.0" alpha="0.0" colorSpace="calibratedWhite"/>
                                 </view>
-                                <constraints>
-                                    <constraint firstItem="pF7-Pf-FWu" firstAttribute="leading" secondItem="p1j-8r-ocZ" secondAttribute="leading" id="Hod-35-2hs"/>
-                                    <constraint firstAttribute="trailing" secondItem="pF7-Pf-FWu" secondAttribute="trailing" id="MN4-qQ-Njf"/>
-                                    <constraint firstItem="pF7-Pf-FWu" firstAttribute="top" secondItem="p1j-8r-ocZ" secondAttribute="top" id="ffr-3v-uYw"/>
-                                    <constraint firstAttribute="bottom" secondItem="pF7-Pf-FWu" secondAttribute="bottom" id="muO-A8-P6o"/>
-                                </constraints>
                                 <size key="customSize" width="200" height="200"/>
                                 <connections>
                                     <outlet property="imageView" destination="pF7-Pf-FWu" id="8T9-kt-qX6"/>
@@ -69,7 +61,6 @@
                             <subviews>
                                 <label opaque="NO" clipsSubviews="YES" userInteractionEnabled="NO" contentMode="left" horizontalHuggingPriority="251" verticalHuggingPriority="251" text="Header" lineBreakMode="tailTruncation" baselineAdjustment="alignBaselines" adjustsFontSizeToFit="NO" translatesAutoresizingMaskIntoConstraints="NO" id="o3C-W5-65K">
                                     <rect key="frame" x="480" y="13" width="65" height="24"/>
-                                    <autoresizingMask key="autoresizingMask" flexibleMaxX="YES" flexibleMaxY="YES"/>
                                     <fontDescription key="fontDescription" type="system" pointSize="20"/>
                                     <color key="textColor" cocoaTouchSystemColor="darkTextColor"/>
                                     <nil key="highlightedColor"/>
@@ -87,7 +78,6 @@
                             <subviews>
                                 <label opaque="NO" clipsSubviews="YES" userInteractionEnabled="NO" contentMode="left" horizontalHuggingPriority="251" verticalHuggingPriority="251" text="Footer" lineBreakMode="tailTruncation" baselineAdjustment="alignBaselines" adjustsFontSizeToFit="NO" translatesAutoresizingMaskIntoConstraints="NO" id="3JB-x1-2zj">
                                     <rect key="frame" x="483" y="13" width="58" height="24"/>
-                                    <autoresizingMask key="autoresizingMask" flexibleMaxX="YES" flexibleMaxY="YES"/>
                                     <fontDescription key="fontDescription" type="system" pointSize="20"/>
                                     <color key="textColor" cocoaTouchSystemColor="darkTextColor"/>
                                     <nil key="highlightedColor"/>
@@ -116,4 +106,4 @@
         <simulatedOrientationMetrics key="orientation"/>
         <simulatedScreenMetrics key="destination"/>
     </simulatedMetricsContainer>
-</document>
\ No newline at end of file
+</document>
diff --git a/NHBalancedFlowLayoutDemo/NHBalancedFlowLayoutDemo/ViewController.m b/NHBalancedFlowLayoutDemo/NHBalancedFlowLayoutDemo/ViewController.m
index 69fcddc..73d5ef3 100644
--- a/NHBalancedFlowLayoutDemo/NHBalancedFlowLayoutDemo/ViewController.m
+++ b/NHBalancedFlowLayoutDemo/NHBalancedFlowLayoutDemo/ViewController.m
@@ -86,6 +86,7 @@ - (UICollectionViewCell *)collectionView:(UICollectionView *)collectionView cell
             NSIndexPath *currentIndexPathForCell = [collectionView indexPathForCell:cell];
             if (currentIndexPathForCell.row == rowIndex) {
                 cell.imageView.image = image;
+                cell.imageView.frame = cell.bounds;
             }
         });
     });