WR.renegade_job = JobVariant(JobPrefab.Get("renegadeteam"), 0)
WR.coalition_job = JobVariant(JobPrefab.Get("coalitionteam"), 0)

Hook.Add('jobsAssigned', 'WR_JobBalance', function()
    local amountCoalition = 0
    local amountRenegade = 0
    for key, value in pairs(Client.ClientList) do
        if value.AssignedJob.Prefab.Identifier == 'renegadeteam' then
            amountRenegade = amountRenegade + 1
        elseif value.AssignedJob.Prefab.Identifier == 'coalitionteam' then
            amountCoalition = amountCoalition + 1
        end
    end

    local difference = amountCoalition - amountRenegade

    print("difference is", difference)

    if difference > 1 then
        local amount = math.abs(difference) - 2
        local deadPlayers = WR.GetDeadPlayers()
        for key, value in pairs(deadPlayers) do
            if value.AssignedJob.Prefab.Identifier == 'coalitionteam' then
                value.AssignedJob = WR.renegade_job
                amount = amount - 1
                if amount == 0 then
                    break
                end
            end
        end
    elseif difference < -1 then
        local amount = math.abs(difference) - 2
        local deadPlayers = WR.GetDeadPlayers()
        for key, value in pairs(deadPlayers) do
            if value.AssignedJob.Prefab.Identifier == 'renegadeteam' then
                value.AssignedJob = WR.coalition_job
                amount = amount - 1
                if amount == 0 then
                    break
                end
            end
        end
    end
end)