diff --git a/src/verify.py b/src/verify.py
index bb8525a2..f8875541 100644
--- a/src/verify.py
+++ b/src/verify.py
@@ -6,13 +6,13 @@
 # The number 3 is [not] present in the array.
 #
 #
-# #include <iostream>
-# using namespace std;
-#
-# int main()
-# {
-#   // placeholder
-#   int N[10] = {3, 4, 5, 1, 2, 3, 4, 9, 13, 0};
-#
-#   return 0;
-# }
+
+
+def contains(n: int) -> bool:
+    arr = [1, 2, 3, 4]
+    return n in arr
+
+
+if __name__ == "__main__":
+    print(contains(1))
+    print(contains(5))
diff --git a/tests/test_verify.py b/tests/test_verify.py
new file mode 100644
index 00000000..dc824a54
--- /dev/null
+++ b/tests/test_verify.py
@@ -0,0 +1,6 @@
+from src.verify import contains
+
+
+def test_contains() -> None:
+    assert contains(5) is False
+    assert contains(1) is True