Updated the app to UICollectionViewController