Hello @nonameroses welcome to the community!
If you auth a user through an organization either by adding the org_id as a param in the authorize request OR by relying on the organization picker built into Universal Login then the org_id is automatically added as a claim to both the access and ID tokens.
Are you also looking to add roles and permissions to tokens?
Let us know!