update generator class to fit our oddball API.