diff --git a/hydra_redis/hydra_graph.py b/hydra_redis/hydra_graph.py
index 11cd1d8..405d370 100644
--- a/hydra_redis/hydra_graph.py
+++ b/hydra_redis/hydra_graph.py
@@ -90,7 +90,7 @@ def main(self,new_url,api_doc):
 
 if __name__ == "__main__":
     url = "http://35.224.198.158:8080/api"
-    apidoc = final_file(url + "/vocab")
-    api_doc = doc_maker.create_doc(apidoc)
     initial_graph = InitialGraph()
+    apidoc = initial_graph.final_file(url + "/vocab")
+    api_doc = doc_maker.create_doc(apidoc)
     initial_graph.main(url, api_doc)