Как указать количество строк, столбцов и интервал для ячейки CollectionView

Question

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

func collectionView(_ collectionView: UICollectionView,
                    layout collectionViewLayout: 
UICollectionViewLayout,
                    sizeForItemAt indexPath: IndexPath) -> CGSize {

    let layout = collectionViewLayout as! UICollectionViewFlowLayout;


    layout.minimumInteritemSpacing = 0.6
    layout.minimumLineSpacing = 0.6

    let remainingInterSpace = collectionView.frame.width - (layout.minimumInteritemSpacing*(numberOfColumns-1))       
    let remainingLineSpace = collectionView.frame.height - (layout.minimumLineSpacing*(numberOfRows-1))


    let width = remainingInterSpace / numberOfColumns     
    let height = remainingLineSpace / numberOfRows


    return CGSize(width: width, height: height)

 }