diff --git a/src/lib.rs b/src/lib.rs
new file mode 100644
index 0000000..3252967
--- /dev/null
+++ b/src/lib.rs
@@ -0,0 +1,3 @@
+pub mod flow;
+pub mod io;
+pub mod types;
diff --git a/tests/integration.rs b/tests/integration.rs
new file mode 100644
index 0000000..c9e4975
--- /dev/null
+++ b/tests/integration.rs
@@ -0,0 +1,14 @@
+use pathfinder2::flow::compute_flow;
+use pathfinder2::io::read_edges_binary;
+use pathfinder2::types::Address;
+
+#[test]
+fn test_flow() {
+    let edges = read_edges_binary(&"edges.dat".to_string()).unwrap();
+    let transfers = compute_flow(
+        &Address::from("0x8DC7e86fF693e9032A0F41711b5581a04b26Be2E"),
+        &Address::from("0x42cEDde51198D1773590311E2A340DC06B24cB37"),
+        &edges,
+    );
+    println!("{:?}", transfers);
+}