Add layoutDelgate and set size of UICollectionViewCells
Bug fixes