diff --git a/cmd/gen_graph/main.go b/cmd/gen_graph/main.go
index 61ee8cd..9d2df12 100644
--- a/cmd/gen_graph/main.go
+++ b/cmd/gen_graph/main.go
@@ -16,7 +16,7 @@ var gzip_files bool
 
 // https://github.com/bmeg/sifter/blob/51a67b0de852e429d30b9371d9975dbefe3a8df9/transform/graph_build.go#L86
 var Cmd = &cobra.Command{
-	Use:   "gen-graph [schema dir] [data dir] [out dir] [class name]",
+	Use:   "gen-graph [schema dir or schema file] [data dir] [out dir] [class name]",
 	Short: "Generates edges and vertices from source data files and schemas",
 	Args:  cobra.MinimumNArgs(4),
 	RunE: func(cmd *cobra.Command, args []string) error {
diff --git a/util/generate.go b/util/generate.go
index 00f0725..a3d3b34 100644
--- a/util/generate.go
+++ b/util/generate.go
@@ -133,12 +133,10 @@ func (s GraphSchema) Generate(classID string, data map[string]any, clean bool, p
 				}
 			}
 			for name := range class.Properties {
-				// gather compare to a list of rels so that the vertexes don't include edge reference information
-				if !contains(ListOfRels, name) {
-					if d, ok := data[name]; ok {
-						vData[name] = d
-					}
+				if d, ok := data[name]; ok {
+					vData[name] = d
 				}
+
 			}
 			if project_id != "" {
 				project_parts := strings.Split(project_id, "-")