Added functionality to the prefab generator