Add comment to generated code