diff --git a/rustfmt.toml b/rustfmt.toml
new file mode 100644
index 0000000..3a26366
--- /dev/null
+++ b/rustfmt.toml
@@ -0,0 +1 @@
+edition = "2021"
diff --git a/src/models/text_content.rs b/src/models/text_content.rs
index 8148ca7..eb86833 100644
--- a/src/models/text_content.rs
+++ b/src/models/text_content.rs
@@ -5,7 +5,7 @@ use crate::{
 };
 use async_openai::error::OpenAIError;
 use serde::{Deserialize, Serialize};
-use surrealdb::{engine::remote::ws::Client, Surreal};
+use surrealdb::{engine::remote::ws::Client, sql::Thing, Surreal};
 use text_splitter::TextSplitter;
 use thiserror::Error;
 use tracing::{debug, info};
@@ -83,7 +83,7 @@ async fn get_related_nodes(
     id: String,
     db_client: &Surreal<Client>,
 ) -> Result<Vec<KnowledgeEntity>, ProcessingError> {
-    let query = format!("SELECT -> knowledge_relationship -> knowledge_entity as related_nodes FROM knowledge_entity WHERE source_id = `{}`", id);
+    let query = format!("SELECT * FROM knowledge_entity WHERE source_id = '{}'", id);
 
     // let query = format!("SELECT * FROM knowledge_entity WHERE in OR out {}", id);
     let related_nodes: Vec<KnowledgeEntity> = db_client.query(query).await?.take(0)?;
@@ -101,7 +101,7 @@ impl TextContent {
         self.store_text_content(&db_client).await?;
 
         let closest_text_content: Vec<TextChunk> = vector_comparison(
-            4,
+            3,
             self.text.clone(),
             &db_client,
             "text_chunk".to_string(),
@@ -110,9 +110,10 @@ impl TextContent {
         .await?;
 
         for node in closest_text_content {
-            info!("{}-{}", node.id, node.source_id);
             let related_nodes = get_related_nodes(node.source_id, &db_client).await?;
-            info!("{:?}", related_nodes);
+            for related_node in related_nodes {
+                info!("{:?}", related_node.name);
+            }
         }
 
         panic!("STOPPING");
diff --git a/src/routes/file.rs b/src/routes/file.rs
index 8da870a..103916c 100644
--- a/src/routes/file.rs
+++ b/src/routes/file.rs
@@ -1,5 +1,4 @@
 use std::sync::Arc;
-
 use axum::{
     extract::Path, response::IntoResponse, Extension, Json
 };
@@ -8,7 +7,6 @@ use serde_json::json;
 use tempfile::NamedTempFile;
 use tracing::info;
 use uuid::Uuid;
-
 use crate::{
     models::file_info::{FileError, FileInfo},
      surrealdb::SurrealDbClient,