diff --git a/Justfile b/Justfile
new file mode 100644
index 0000000..dd2a448
--- /dev/null
+++ b/Justfile
@@ -0,0 +1,10 @@
+alias b := bench
+
+store:
+    cargo bench -- --save-baseline base
+
+bench:
+    cargo bench -- --baseline base
+
+flame:
+    cargo flamegraph --root --example atp
diff --git a/graph.d2 b/graph.d2
deleted file mode 100644
index bf8267b..0000000
--- a/graph.d2
+++ /dev/null
@@ -1,64 +0,0 @@
-vars: {
-  d2-config: {
-    layout-engine: elk
-    # Terminal theme code
-    theme-id: 300
-  }
-}
-
-History: {
-  shape: class
-
-  agents: "HashMap<Index, Agent>"
-  batches: "Vec<Batch>"
-}
-
-Batch: {
-  shape: class
-
-  skills: "HashMap<Index, Skill>"
-  events: "Vec<Event>"
-  time: "i64"
-  p_draw: "f64"
-}
-
-Event: {
-  shape: class
-
-  teams: "Vec<Team>"
-  weights: "Vec<Vec<f64>>"
-  evidence: "f64"
-}
-
-Team: {
-  shape: class
-
-  items: "Vec<Item>"
-  output: "f64"
-}
-
-Item: {
-  shape: class
-
-  agent: "Index"
-  likelihood: "Gaussian"
-}
-
-Skill: {
-  shape: class
-
-  forward: "Gaussian"
-  backward: "Gaussian"
-  likelihood: "Gaussian"
-  elapsed: "i64"
-  online: "Gaussian"
-}
-
-History -> Batch
-
-Batch -> Skill
-Batch -> Event
-
-Event -> Team
-
-Team -> Item