Add state parameter for user role assignment