Add check for team membership