add generator class