X-Git-Url: http://www2.svjatoslav.eu/gitweb/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Feu%2Fsvjatoslav%2Finspector%2Fjava%2Fstructure%2Fexample%2FRenderDemoClasses.java;h=90aa4800ece8c25c2e00d5ef24b7fcf7aa45ceee;hb=c680c10ad9057106e23b149246abdd84b41775ee;hp=9c4815a958af929d79ddb1b3332ab4c7f7c0f5b9;hpb=9d9b65fb1909e12606e76ed0625706e446f588fb;p=javainspect.git

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 9c4815a..90aa480 100755
--- a/src/test/java/eu/svjatoslav/inspector/java/structure/example/RenderDemoClasses.java
+++ b/src/test/java/eu/svjatoslav/inspector/java/structure/example/RenderDemoClasses.java
@@ -1,7 +1,7 @@
 /*
  * JavaInspect - Utility to visualize java software
  * Copyright (C) 2013-2015, Svjatoslav Agejenko, svjatoslav@svjatoslav.eu
- * 
+ *
  * This program is free software; you can redistribute it and/or
  * modify it under the terms of version 3 of the GNU Lesser General Public License
  * or later as published by the Free Software Foundation.
@@ -11,15 +11,14 @@ package eu.svjatoslav.inspector.java.structure.example;
 
 import eu.svjatoslav.inspector.java.structure.ClassGraph;
 import eu.svjatoslav.inspector.java.structure.example.structure.SampleClass;
+import eu.svjatoslav.inspector.java.structure.example.structure.SampleClass2;
 
 public class RenderDemoClasses {
 
-	public static void main(final String[] args) {
-		final ClassGraph graph = new ClassGraph();
+    public static void main(final String[] args) {
 
-		graph.addClass(SampleClass.class);
-
-		graph.generateGraph("example");
-	}
+        new ClassGraph().add(SampleClass.class, SampleClass2.class)
+                .generateGraph("example");
+    }
 
 }