Moved logic to subclass.