Add argument to factory classmethod