Hey @jjones1 - Really sorry I never got back to you on this one, slipped through the cracks somehow 
For future reference, the code looks good to me - I would however like to point out that the Management API is rate limited so be aware of that.
Here’s an FAQ that may be of use as well:
https://community.auth0.com/t/how-to-add-roles-and-permissions-to-the-id-token-using-actions/84506
Cheers!