Как показать коллекцию, как это, используя swift 4?

Question

Здравствуйте, я хочу показать несколько изображений в collectionView, как на картинке выше. я могу показать две строки и три столбца с помощью UICollectionViewDelegateFlowLayout, но не могу этого сделать. помощь очень ценится.

func collectionView(_ collectionView: UICollectionView, layout collectionViewLayout: UICollectionViewLayout, sizeForItemAt indexPath: IndexPath) -> CGSize {        
        switch indexPath.item {
        case 0,1:
            return CGSize(width: (UIScreen.main.bounds.width - 4) / 2, height: (UIScreen.main.bounds.width - 4) / 2)
        default:
            return CGSize(width: (UIScreen.main.bounds.width - 6) / 3, height:  (UIScreen.main.bounds.width) / 3)
        }

    }