diff --git a/helper.py b/helper.py
index df6d2cf..62cbac3 100644
--- a/helper.py
+++ b/helper.py
@@ -75,12 +75,14 @@ def to_state_string(state, state_nex=None):
 
     return state_str
 
-print("Testing")
-test_sample([0.7, 0.1, 0.1, 0.1])
-test_sample([0.1, 0.1, 0.2, 0.1])
-test_sample([0.2, 0.8])
-test_sample([0.1, 0.1])
-test_sample([0.3, 0.0, 0.2, 0.0])
-test_sample([0.0, 0.0, 0.0, 0.0])
-test_sample([0.5, 0.0, 0.2, 0.3])
-test_sample([0.3, 0.1, 0.4, 0.2])
+
+if __name__ == '__main__':
+    print("Testing sample()")
+    test_sample([0.7, 0.1, 0.1, 0.1])
+    test_sample([0.1, 0.1, 0.2, 0.1])
+    test_sample([0.2, 0.8])
+    test_sample([0.1, 0.1])
+    test_sample([0.3, 0.0, 0.2, 0.0])
+    test_sample([0.0, 0.0, 0.0, 0.0])
+    test_sample([0.5, 0.0, 0.2, 0.3])
+    test_sample([0.3, 0.1, 0.4, 0.2])