diff --git a/src/modules/generationtools/modelgeneration.py b/src/modules/generationtools/modelgeneration.py
index 76577b9..1bc66a5 100644
--- a/src/modules/generationtools/modelgeneration.py
+++ b/src/modules/generationtools/modelgeneration.py
@@ -84,10 +84,11 @@ def findCovariances(df, dists, params):
     cos = np.cov(data)
     return cos
 
-'''
-Sample use:
+def main():
+    iris = pd.read_csv('Iris.csv')
+    iris = iris.drop(columns=['Species'])
+    b, pv, p = findBestDistribution(iris)
+    print(b)
 
-data = pd.read_csv("test.csv")
-dists, pvalues, params = findBestDistribution(data)
-cos = findCovariances(data, dists, params)
-'''
\ No newline at end of file
+if __name__ == '__main__':
+    main()
\ No newline at end of file