To fix the issue where necessary information is missing from the claims in the previous test results, in this section we will add a Cognito Extension to intervene before the token is generated.