integrate Generator class to plugin