function Add-OverflowCustomAttribute {
<#
.SYNOPSIS
    Adds Overflow: true or false as custom node attributes in newrelic-infra.yml

.DESCRIPTION
    If a server is an overflow server, Perf wants to know via new relic custom attribute.
    This affects the server that this function is being run on.

.PARAMETER IsOverflow
    IsOverflow sets the value to true or false

.EXAMPLE
    Add-OverflowCustomAttribute -IsOverflow $true
#>

    [CmdletBinding()]
    Param(
        [Parameter(Mandatory = $true)]
        [bool]$IsOverflow
    )

    $logLead = Get-LogLeadName

    $isDirty = $false
    Import-Module powershell-yaml

    $Path = Get-NewRelicYamlPath
    # Returning/stopping exactly where the error is
    if (Test-StringIsNullOrWhitespace $Path) { return }

    $fileContent = Get-Content $Path -Raw
    $yaml = ConvertFrom-Yaml -Yaml $fileContent -Ordered

    # This will add the custom_attribute field if it isn't already present
    # Otherwise it will update the Overflow value
    if (!($yaml.custom_attributes)) {
        $yaml.custom_attributes = @{Overflow = $IsOverflow }
        $isDirty = $true
    } elseif ($yaml.custom_attributes.Overflow -ne $IsOverflow) {
        $yaml.custom_attributes.Overflow = $IsOverflow
        $isDirty = $true
    }

    if ($isDirty) {
        Write-Host "$loglead : Setting Overflow Attribute to $IsOverflow"
        Write-Host "$loglead : Saving to path: $Path"
        ConvertTo-Yaml -Data $yaml -Outfile $Path -Force
    } else {
        Write-Host "$loglead : Nothing to change in the config file"
    }

}