2 files changed, 12 insertions, 8 deletions
diff --git a/examples/08-custom-code/output/rust/src/lib.rs b/examples/08-custom-code/output/rust/src/lib.rs
new file mode 100644
index 0000000..1825d72
--- /dev/null
+++ b/examples/08-custom-code/output/rust/src/lib.rs
@@ -0,0 +1,12 @@
+mod shapes;
+
+#[cfg(test)]
+mod test {
+    use super::shapes::Square;
+
+    #[test]
+    fn can_use_custom_code() {
+        let square = Square { width: 10 };
+        assert_eq!(square.area(), 100);
+    }
+}
diff --git a/examples/08-custom-code/output/rust/src/main.rs b/examples/08-custom-code/output/rust/src/main.rs
deleted file mode 100644
index e30443e..0000000
--- a/examples/08-custom-code/output/rust/src/main.rs
+++ /dev/null
@@ -1,8 +0,0 @@
-mod shapes;
-
-use shapes::Square;
-
-fn main() {
-    let square = Square { width: 10 };
-    assert!(square.area() == 100);
-}