Some additional adjustments to the prefab generator