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=9720b6ec40d9b10e5a2df8fad848b82f36451c1b;hp=7e9e5a3812f4bdc261de9cca9872fd9e7b4f9526;hpb=31e8e0b97c9f60f8820d708fa86c11d65f5445f3;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 7e9e5a3..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-2014, Svjatoslav Agejenko, svjatoslav@svjatoslav.eu
- * 
+ * 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");
+    }
 
 }