@@ -2,8 +2,8 @@  | 
                                                    ||
| 2 | 2 | |
| 3 | 3 | namespace BeyondCode\ErdGenerator;  | 
                                                        
| 4 | 4 | |
| 5 | -use phpDocumentor\GraphViz\Graph;  | 
                                                        |
| 6 | 5 | use Illuminate\Support\Collection;  | 
                                                        
| 6 | +use phpDocumentor\GraphViz\Graph;  | 
                                                        |
| 7 | 7 | use phpDocumentor\GraphViz\Node;  | 
                                                        
| 8 | 8 | |
| 9 | 9 | class GraphBuilder  | 
                                                        
@@ -2,11 +2,11 @@  | 
                                                    ||
| 2 | 2 | |
| 3 | 3 | namespace BeyondCode\ErdGenerator;  | 
                                                        
| 4 | 4 | |
| 5 | -use ReflectionClass;  | 
                                                        |
| 5 | +use BeyondCode\ErdGenerator\Model as GraphModel;  | 
                                                        |
| 6 | 6 | use Illuminate\Console\Command;  | 
                                                        
| 7 | -use phpDocumentor\GraphViz\Graph;  | 
                                                        |
| 8 | 7 | use Illuminate\Support\Collection;  | 
                                                        
| 9 | -use BeyondCode\ErdGenerator\Model as GraphModel;  | 
                                                        |
| 8 | +use ReflectionClass;  | 
                                                        |
| 9 | +use phpDocumentor\GraphViz\Graph;  | 
                                                        |
| 10 | 10 | |
| 11 | 11 | class GenerateDiagramCommand extends Command  | 
                                                        
| 12 | 12 |  { | 
                                                        
@@ -2,14 +2,12 @@  | 
                                                    ||
| 2 | 2 | |
| 3 | 3 | namespace BeyondCode\ErdGenerator;  | 
                                                        
| 4 | 4 | |
| 5 | -use ReflectionClass;  | 
                                                        |
| 6 | -use ReflectionMethod;  | 
                                                        |
| 7 | -use Illuminate\Support\Collection;  | 
                                                        |
| 8 | -use Illuminate\Database\Eloquent\Relations\Relation;  | 
                                                        |
| 9 | 5 | use Illuminate\Database\Eloquent\Relations\BelongsTo;  | 
                                                        
| 10 | -use Illuminate\Database\Eloquent\Relations\MorphToMany;  | 
                                                        |
| 11 | 6 | use Illuminate\Database\Eloquent\Relations\HasOneOrMany;  | 
                                                        
| 12 | -use Illuminate\Database\Eloquent\Relations\BelongsToMany;  | 
                                                        |
| 7 | +use Illuminate\Database\Eloquent\Relations\Relation;  | 
                                                        |
| 8 | +use Illuminate\Support\Collection;  | 
                                                        |
| 9 | +use ReflectionClass;  | 
                                                        |
| 10 | +use ReflectionMethod;  | 
                                                        |
| 13 | 11 | |
| 14 | 12 | class RelationFinder  | 
                                                        
| 15 | 13 |  { | 
                                                        
@@ -2,14 +2,14 @@  | 
                                                    ||
| 2 | 2 | |
| 3 | 3 | namespace BeyondCode\ErdGenerator;  | 
                                                        
| 4 | 4 | |
| 5 | +use Illuminate\Database\Eloquent\Model as EloquentModel;  | 
                                                        |
| 6 | +use Illuminate\Filesystem\Filesystem;  | 
                                                        |
| 7 | +use Illuminate\Support\Collection;  | 
                                                        |
| 5 | 8 | use PhpParser\NodeTraverser;  | 
                                                        
| 6 | -use PhpParser\ParserFactory;  | 
                                                        |
| 9 | +use PhpParser\NodeVisitor\NameResolver;  | 
                                                        |
| 7 | 10 | use PhpParser\Node\Stmt\Class_;  | 
                                                        
| 8 | 11 | use PhpParser\Node\Stmt\Namespace_;  | 
                                                        
| 9 | -use Illuminate\Support\Collection;  | 
                                                        |
| 10 | -use Illuminate\Filesystem\Filesystem;  | 
                                                        |
| 11 | -use PhpParser\NodeVisitor\NameResolver;  | 
                                                        |
| 12 | -use Illuminate\Database\Eloquent\Model as EloquentModel;  | 
                                                        |
| 12 | +use PhpParser\ParserFactory;  | 
                                                        |
| 13 | 13 | |
| 14 | 14 | class ModelFinder  | 
                                                        
| 15 | 15 |  { |