local Workspace = game:GetService("Workspace")

local folders = {
	Workspace:WaitForChild("TripMines"),
	Workspace:WaitForChild("Traps")
}

-- Create 2D billboard
local function addBillboard(model)
	if not model:IsA("Model") then return end
	
	-- Prevent duplicates
	if model:FindFirstChild("TrapBillboard") then return end
	
	local basePart = model:FindFirstChildWhichIsA("BasePart", true)
	if not basePart then return end
	
	-- BillboardGui
	local billboard = Instance.new("BillboardGui")
	billboard.Name = "TrapBillboard"
	billboard.Size = UDim2.new(0, 40, 0, 40)
	billboard.StudsOffset = Vector3.new(0, 2.5, 0)
	billboard.AlwaysOnTop = true
	billboard.Parent = basePart
	
	-- Red circle
	local circle = Instance.new("Frame")
	circle.Size = UDim2.new(1, 0, 1, 0)
	circle.BackgroundColor3 = Color3.fromRGB(255, 0, 0)
	circle.BorderSizePixel = 0
	circle.Parent = billboard
	
	-- Make it circular
	local corner = Instance.new("UICorner")
	corner.CornerRadius = UDim.new(1, 0)
	corner.Parent = circle
end

-- Connect folders
for _, folder in ipairs(folders) do
	
	-- Existing models
	for _, child in ipairs(folder:GetChildren()) do
		addBillboard(child)
	end
	
	-- New models
	folder.ChildAdded:Connect(function(child)
		task.wait()
		addBillboard(child)
	end)
	
end