Enable code generator more customizable