diff --git a/convention/src/main/kotlin/io/gitlab/jfronny/scripts/CodegenExt.kt b/convention/src/main/kotlin/io/gitlab/jfronny/scripts/CodegenExt.kt
index 2a62dc0..6a0c0bb 100644
--- a/convention/src/main/kotlin/io/gitlab/jfronny/scripts/CodegenExt.kt
+++ b/convention/src/main/kotlin/io/gitlab/jfronny/scripts/CodegenExt.kt
@@ -152,3 +152,7 @@ fun ParameterSpec.Builder.javadoc(generate: Action<CodeBlock.Builder>) = addJava
 fun ParameterSpec.Builder.annotation(annotation: Class<*>, generate: Action<AnnotationSpec.Builder>? = null) = addAnnotation(AnnotationSpec.builder(annotation).run(generate).build())
 fun ParameterSpec.Builder.annotation(annotation: ClassName, generate: Action<AnnotationSpec.Builder>? = null) = addAnnotation(AnnotationSpec.builder(annotation).run(generate).build())
 fun ParameterSpec.Builder.modifiers(vararg modifiers: Modifier) = addModifiers(*modifiers)
+
+// Extensions for CodeBlock.Builder
+fun CodeBlock.Builder.controlFlow(controlFlow: String, vararg args: Any, generate: Action<CodeBlock.Builder>) = beginControlFlow(controlFlow, *args).run(generate).endControlFlow()
+fun CodeBlock.Builder.indent(generate: Action<CodeBlock.Builder>) = indent().run(generate).unindent()