Генерация массива Nested Sets

Есть рабочая функция на Yii2, которая принимает на вход количество узлов в дереве и на основе этого параметра генерирует массив виде дерева - Nested Sets и инсертит его в БД.

    public static function generationTree($componentGeneration){
    $tree = new Tree();
    $tree->deleteAll();
    $parentEl = 0;
    for($i=1; $i<=$componentGeneration*$componentGeneration; $i++){
        $tree = new Tree(['id' => "$i", 'name' => "$i", 'url' => "$i"]);
        $tree->parent = $parentEl;
        if($tree->parent == 0){
            $tree->makeRoot();
        }
        else{
            if($tree->parent > $componentGeneration){
                $q = Tree::find()->andWhere(['id' => rand(1,$tree->parent)])->one();
            }
            else{
                $q = Tree::find()->andWhere(['id' => $tree->parent])->one();
            }
            $tree->prependTo($q);
        }
        $parentEl++;
    }
    $tree->save();
}

Подскажите плиз как эту функцию по уму переписать?