added class for code generation