Shield - 2FA for only some users

Hi,
Nevermind, I figured it out.
Thank you for your help @kenjis

Anyone who comes across this.
I added to my users table a column called email2fa
*Either 0 or 1*, 1 being they get 2fa.

Extended the Email2FA class

Override the createIdentify function as follows works:

Code:

public function createIdentity(User $user): string
    {
        if($user->email2fa != '1'){
            return '';
        }
        /** @var UserIdentityModel $identityModel */
        $identityModel = model(UserIdentityModel::class);

        // Delete any previous identities for action
        $identityModel->deleteIdentitiesByType($user, $this->type);

        $generator = static fn (): string => random_string('nozero', 6);

        return $identityModel->createCodeIdentity(
            $user,
            [
                'type'  => $this->type,
                'name'  => 'login',
                'extra' => lang('Auth.need2FA'),
            ],
            $generator
        );
    }

Sorry for this newbie question. How do I extend this class? Currently, I directly added the code snippet to the actual file.