1 files changed, 16 insertions, 17 deletions
diff --git a/javaparser-core/src/main/java/com/github/javaparser/ast/CompilationUnit.java b/javaparser-core/src/main/java/com/github/javaparser/ast/CompilationUnit.java
index 485558a56..dcf68e294 100644
--- a/javaparser-core/src/main/java/com/github/javaparser/ast/CompilationUnit.java
+++ b/javaparser-core/src/main/java/com/github/javaparser/ast/CompilationUnit.java
@@ -21,25 +21,21 @@
 
 package com.github.javaparser.ast;
 
-import static com.github.javaparser.utils.Utils.ensureNotNull;
+import com.github.javaparser.Range;
+import com.github.javaparser.ast.body.*;
+import com.github.javaparser.ast.comments.Comment;
+import com.github.javaparser.ast.comments.JavadocComment;
+import com.github.javaparser.ast.expr.NameExpr;
+import com.github.javaparser.ast.visitor.GenericVisitor;
+import com.github.javaparser.ast.visitor.VoidVisitor;
+import com.github.javaparser.utils.ClassUtils;
 
 import java.util.Arrays;
 import java.util.EnumSet;
 import java.util.List;
 import java.util.stream.Collectors;
 
-import com.github.javaparser.ast.expr.NameExpr;
-import com.github.javaparser.utils.ClassUtils;
-import com.github.javaparser.Range;
-import com.github.javaparser.ast.body.AnnotationDeclaration;
-import com.github.javaparser.ast.body.ClassOrInterfaceDeclaration;
-import com.github.javaparser.ast.body.EmptyTypeDeclaration;
-import com.github.javaparser.ast.body.EnumDeclaration;
-import com.github.javaparser.ast.body.TypeDeclaration;
-import com.github.javaparser.ast.comments.Comment;
-import com.github.javaparser.ast.comments.JavadocComment;
-import com.github.javaparser.ast.visitor.GenericVisitor;
-import com.github.javaparser.ast.visitor.VoidVisitor;
+import static com.github.javaparser.utils.Utils.ensureNotNull;
 
 /**
  * <p>
@@ -152,7 +148,7 @@ public final class CompilationUnit extends Node {
      * @param comments
      *            the list of comments
      */
-    public void setComments(List<Comment> comments) {
+    public CompilationUnit setComments(List<Comment> comments) {
         throw new RuntimeException("Not implemented!");
     }
 
@@ -163,9 +159,10 @@ public final class CompilationUnit extends Node {
      * @param imports
      *            the list of imports
      */
-    public void setImports(List<ImportDeclaration> imports) {
+    public CompilationUnit setImports(List<ImportDeclaration> imports) {
         this.imports = imports;
         setAsParentNodeOf(this.imports);
+        return this;
     }
 
     /**
@@ -175,9 +172,10 @@ public final class CompilationUnit extends Node {
      *            the pakage declaration to set or <code>null</code> to default
      *            package
      */
-    public void setPackage(PackageDeclaration pakage) {
+    public CompilationUnit setPackage(PackageDeclaration pakage) {
         this.pakage = pakage;
         setAsParentNodeOf(this.pakage);
+        return this;
     }
 
     /**
@@ -186,9 +184,10 @@ public final class CompilationUnit extends Node {
      * @param types
      *            the lis of types
      */
-    public void setTypes(List<TypeDeclaration<?>> types) {
+    public CompilationUnit setTypes(List<TypeDeclaration<?>> types) {
         this.types = types;
         setAsParentNodeOf(this.types);
+        return this;
     }
 
     /**