diff --git a/src/main.rs b/src/main.rs
index 960adba..a6f1046 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,9 +1,8 @@
 pub mod args;
 pub mod child;
-pub mod optimised_trace;
 pub mod parent;
-pub mod raw_trace;
 pub mod tui;
+pub mod trace;
 
 use args::Args;
 use child::child;
diff --git a/src/trace/mod.rs b/src/trace/mod.rs
new file mode 100644
index 0000000..6af6dfd
--- /dev/null
+++ b/src/trace/mod.rs
@@ -0,0 +1,2 @@
+pub mod optimised_trace;
+pub mod raw_trace;
\ No newline at end of file
diff --git a/src/optimised_trace.rs b/src/trace/optimised_trace.rs
similarity index 98%
rename from src/optimised_trace.rs
rename to src/trace/optimised_trace.rs
index 789dead..77d8434 100644
--- a/src/optimised_trace.rs
+++ b/src/trace/optimised_trace.rs
@@ -1,6 +1,6 @@
 use std::{collections::HashSet, hash::Hash};
 
-use crate::raw_trace::{RawTrace, InstructionPointer, Timestamp};
+use crate::trace::raw_trace::{RawTrace, InstructionPointer, Timestamp};
 
 #[derive(Debug)]
 pub struct StackFrameID(u64);
diff --git a/src/raw_trace.rs b/src/trace/raw_trace.rs
similarity index 100%
rename from src/raw_trace.rs
rename to src/trace/raw_trace.rs
diff --git a/src/tui.rs b/src/tui.rs
index c68deef..df68178 100644
--- a/src/tui.rs
+++ b/src/tui.rs
@@ -1,5 +1,4 @@
-use crate::{args::Args, optimised_trace::OptimisedTrace};
-use crate::raw_trace::RawTrace;
+use crate::{args::Args, trace::{raw_trace::RawTrace, optimised_trace::OptimisedTrace}};
 
 use crossterm::{terminal::{enable_raw_mode, EnterAlternateScreen, disable_raw_mode, LeaveAlternateScreen}, execute, event::{self, Event::Key, KeyCode, KeyModifiers}};
 use std::time::Duration;