Implement hook_form_alter instead of class extension

Problem/Motivation

Currently base class is extended to override the form element, which can cause issues for other modules.

Proposed resolution

We can use hook_form_alter to alter the form elements.