changed imports in class generateor