diff --git a/src/server/http.rs b/src/http_server.rs
similarity index 96%
rename from src/server/http.rs
rename to src/http_server.rs
index dcfe4b3..e89a50b 100644
--- a/src/server/http.rs
+++ b/src/http_server.rs
@@ -11,7 +11,7 @@ use tower_http::limit::RequestBodyLimitLayer;
 use crate::{
     Config, JoecalState,
     discovery::register_device,
-    transfer::upload::{register_prepare_upload, register_upload},
+    transfer::{register_prepare_upload, register_upload},
 };
 
 impl JoecalState {
diff --git a/src/lib.rs b/src/lib.rs
index f259ec8..117eda6 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,7 +1,7 @@
 pub mod discovery;
 pub mod error;
+pub mod http_server;
 pub mod models;
-pub mod server;
 pub mod transfer;
 
 use std::{
@@ -10,11 +10,10 @@ use std::{
     sync::Arc,
 };
 
+use models::device::DeviceInfo;
 use serde::{Deserialize, Serialize};
 use tokio::{net::UdpSocket, sync::Mutex, task::JoinHandle};
-use transfer::session::Session;
-
-use crate::models::device::DeviceInfo;
+use transfer::Session;
 
 pub const DEFAULT_PORT: u16 = 53317;
 pub const MULTICAST_IP: Ipv4Addr = Ipv4Addr::new(224, 0, 0, 167);
diff --git a/src/server/mod.rs b/src/server/mod.rs
deleted file mode 100644
index 3883215..0000000
--- a/src/server/mod.rs
+++ /dev/null
@@ -1 +0,0 @@
-pub mod http;
diff --git a/src/transfer/upload.rs b/src/transfer.rs
similarity index 95%
rename from src/transfer/upload.rs
rename to src/transfer.rs
index 836d6a4..9c4f8bc 100644
--- a/src/transfer/upload.rs
+++ b/src/transfer.rs
@@ -15,9 +15,28 @@ use crate::{
     JoecalState,
     error::{LocalSendError, Result},
     models::{device::DeviceInfo, file::FileMetadata},
-    transfer::session::{Session, SessionStatus},
 };
 
+#[derive(Deserialize, Serialize)]
+pub struct Session {
+    pub session_id: String,
+    pub files: HashMap<String, FileMetadata>,
+    pub file_tokens: HashMap<String, String>,
+    pub receiver: DeviceInfo,
+    pub sender: DeviceInfo,
+    pub status: SessionStatus,
+    pub addr: SocketAddr,
+}
+
+#[derive(PartialEq, Deserialize, Serialize)]
+pub enum SessionStatus {
+    Pending,
+    Active,
+    Completed,
+    Failed,
+    Cancelled,
+}
+
 #[derive(Debug, Serialize, Deserialize)]
 #[serde(rename_all = "camelCase")]
 pub struct PrepareUploadResponse {
diff --git a/src/transfer/download.rs b/src/transfer/download.rs
deleted file mode 100644
index 8b13789..0000000
--- a/src/transfer/download.rs
+++ /dev/null
@@ -1 +0,0 @@
-
diff --git a/src/transfer/mod.rs b/src/transfer/mod.rs
deleted file mode 100644
index b27bc94..0000000
--- a/src/transfer/mod.rs
+++ /dev/null
@@ -1,3 +0,0 @@
-pub mod download;
-pub mod session;
-pub mod upload;
diff --git a/src/transfer/session.rs b/src/transfer/session.rs
deleted file mode 100644
index 25939d7..0000000
--- a/src/transfer/session.rs
+++ /dev/null
@@ -1,25 +0,0 @@
-use std::{collections::HashMap, net::SocketAddr};
-
-use serde::{Deserialize, Serialize};
-
-use crate::models::{device::DeviceInfo, file::FileMetadata};
-
-#[derive(Deserialize, Serialize)]
-pub struct Session {
-    pub session_id: String,
-    pub files: HashMap<String, FileMetadata>,
-    pub file_tokens: HashMap<String, String>,
-    pub receiver: DeviceInfo,
-    pub sender: DeviceInfo,
-    pub status: SessionStatus,
-    pub addr: SocketAddr,
-}
-
-#[derive(PartialEq, Deserialize, Serialize)]
-pub enum SessionStatus {
-    Pending,
-    Active,
-    Completed,
-    Failed,
-    Cancelled,
-}