Code refactoring. Simplified API.

[javainspect.git] / src / test / java / eu / svjatoslav / inspector / java / structure / example / RenderDemoClasses.java

diff --git a/src/test/java/eu/svjatoslav/inspector/java/structure/example/RenderDemoClasses.java b/src/test/java/eu/svjatoslav/inspector/java/structure/example/RenderDemoClasses.java
index b5a24a8..50f9c7e 100755 (executable)
--- a/src/test/java/eu/svjatoslav/inspector/java/structure/example/RenderDemoClasses.java
+++ b/src/test/java/eu/svjatoslav/inspector/java/structure/example/RenderDemoClasses.java
@@ -16,12 +16,9 @@ import eu.svjatoslav.inspector.java.structure.example.structure.SampleClass2;
 public class RenderDemoClasses {
 
        public static void main(final String[] args) {
-               final ClassGraph graph = new ClassGraph();
 
-               graph.addClass(SampleClass.class);
-               graph.addClass(SampleClass2.class);
-
-               graph.generateGraph("example", false);
+               new ClassGraph().add(SampleClass.class, SampleClass2.class)
+                               .generateGraph("example", false);
        }
 
 }