diff --git a/study.py b/study.py
index 2b2aab6..2532d13 100644
--- a/study.py
+++ b/study.py
@@ -63,6 +63,5 @@ def study():
         vis = Visualizer(execID)
         vis.plotHeatmaps()
 
-
-study()
-
+if __name__ == "__main__":
+    study()