From f3ebe405571947eb3a7d1c08d0d64dbc77b2c27b Mon Sep 17 00:00:00 2001
From: Daniel McCrevan <danielmccrevan@gmail.com>
Date: Wed, 7 Nov 2018 16:50:21 -0500
Subject: [PATCH] Merged

---
 src/modules/generationtools/modelgeneration.py | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

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