diff --git a/.github/workflows/generate.yml b/.github/workflows/generate.yml
index 40ab03c6..e6ab7b06 100644
--- a/.github/workflows/generate.yml
+++ b/.github/workflows/generate.yml
@@ -15,6 +15,10 @@ on:
         description: 'The API endpoints for which to generate code'
   workflow_call:
     inputs:
+      packages:
+        required: true
+        type: string
+        description: 'A comma-separated list of packages to generate for which to generate code'
       apiEndpoints:
         required: true
         type: string
@@ -71,29 +75,22 @@ jobs:
         env:
           NEW_RELIC_API_KEY: ${{ secrets.NEW_RELIC_API_KEY }}
 
-      # - name: Create new branch, commit, and push changes
-      #   shell: bash
-      #   run: |
-      #     branch_name="tmp/generated-code"
-
-      #     if git ls-remote --exit-code --heads origin $branch_name > /dev/null; then
-      #       git push origin --delete $branch_name
-      #     fi
-
-      #     git checkout -b $branch_name
-
-      #     git status
-      #     git add pkg .tutone.tmp.yml
-      #     git status
-
-      #     git commit --no-verify -m 'feat(codegen): generate code based on latest API changes'
-      #     git push origin $branch_name
+      - name: Remove old automation branch if present
+        shell: bash
+        run: |
+          branch_name="automation/generated-code"
+          if git ls-remote --exit-code --heads origin $branch_name > /dev/null; then
+            git push origin --delete $branch_name
+          fi
 
       - name: Create pull request
         uses: peter-evans/create-pull-request@v3
         with:
           base: main
-          add-paths: 'pkg'
+          add-paths: |
+            pkg
+            .tutone.tmp.yml
+            types.go
           author: 'nr-developer-toolkit <nr-developer-toolkit@users.noreply.github.com>'
           token: ${{ secrets.DEV_TOOLKIT_TOKEN }}
           commit-message: 'feat(automation): update generated code'
@@ -101,7 +98,17 @@ jobs:
           branch: automation/generated-code
           title: 'feat(automation): generate code based on latest API changes'
           body: |
-            Update generated code using Tutone
+            ### Automated Pull Request
+
+            The following packages have been added or updated: ${{ inputs.packages }}.
+
+            Please perform the following checklist before merging:
+            - [ ] Review the changes to the generated code
+            - [ ] Update applicable tests as necessary
+            - [ ] Double check the .tutone.yml config file for accuracy
+
+            **Note:** If something looks off or you have questions, please reach out to an Observability as Code team member for assistance.
           labels: |
             enhancement
+            automated pr
           draft: true
diff --git a/.github/workflows/graphql-schema.yml b/.github/workflows/graphql-schema.yml
index 9becc4f0..5649ad42 100644
--- a/.github/workflows/graphql-schema.yml
+++ b/.github/workflows/graphql-schema.yml
@@ -15,6 +15,7 @@ jobs:
     runs-on: ubuntu-latest
     outputs:
       apiEndpoints: ${{ steps.schema-diff.outputs.tutone_config }}
+      packages: ${{ steps.schema-diff.outputs.packages }}
     steps:
       - name: Install Node
         uses: actions/setup-node@v4
@@ -161,8 +162,6 @@ jobs:
     uses: newrelic/newrelic-client-go/.github/workflows/generate.yml@feat/automated-codegen
     with:
       apiEndpoints: ${{ needs.checkForApiUpdates.outputs.apiEndpoints }}
+      packages: ${{ needs.checkForApiUpdates.outputs.packages }}
     secrets: inherit
-      # secrets:
-    #   NEW_RELIC_API_KEY: ${{ secrets.NEW_RELIC_API_KEY }}
-    #   DEV_TOOLKIT_TOKEN: ${{ secrets.DEV_TOOLKIT_TOKEN }}
 
diff --git a/schema-test-new.json b/schema-test-new.json
index 9af0499f..3e78b9bb 100644
--- a/schema-test-new.json
+++ b/schema-test-new.json
@@ -24,45 +24,6 @@
             }
           }
         ]
-      },
-      {
-        "name": "userManagementCreateGroup",
-        "description": "A mutation for creating a group in an authentication domain.",
-        "type": {
-          "name": "UserManagementCreateGroupPayload",
-          "kind": "OBJECT"
-        },
-        "args": [
-          {
-            "name": "createGroupOptions",
-            "description": "The input object representing the group to create",
-            "type": {
-              "name": "UserManagementCreateGroup",
-              "kind": "INPUT_OBJECT"
-            }
-          }
-        ]
-      },
-      {
-        "name": "userManagementCreateUser",
-        "description": "A mutation for creating a user in an authentication domain.",
-        "type": {
-          "name": "UserManagementCreateUserPayload",
-          "kind": "OBJECT"
-        },
-        "args": [
-          {
-            "name": "createUserOptions",
-            "description": "The input object representing the user to be created.",
-            "type": {
-              "kind": "NON_NULL",
-              "ofType": {
-                "name": "UserManagementCreateUser",
-                "kind": "INPUT_OBJECT"
-              }
-            }
-          }
-        ]
       }
     ]
   },