Change parrent class to <UICollectionViewFlowLayout>