Пользовательский макет в виде коллекции

Question

Возможен несколько разделов в uicollectionview Pinterest Layout. В классе Pinterest Layout работает несколько разделов. Я пробовал это с помощью класса Pinterest Layout. Пожалуйста, помогите.

func numberOfSections(in collectionView: UICollectionView) -> Int {
        return 2
    }

    func collectionView(_ collectionView: UICollectionView, numberOfItemsInSection section: Int) -> Int {
        if section == 0 {
            return 4
        }else {
            return 20
        }
    }