diff --git a/.idea/.gitignore b/.idea/.gitignore
new file mode 100644
index 00000000..26d33521
--- /dev/null
+++ b/.idea/.gitignore
@@ -0,0 +1,3 @@
+# Default ignored files
+/shelf/
+/workspace.xml
diff --git a/.idea/inspectionProfiles/Project_Default.xml b/.idea/inspectionProfiles/Project_Default.xml
new file mode 100644
index 00000000..447a1e9c
--- /dev/null
+++ b/.idea/inspectionProfiles/Project_Default.xml
@@ -0,0 +1,17 @@
+<component name="InspectionProjectProfileManager">
+  <profile version="1.0">
+    <option name="myName" value="Project Default" />
+    <inspection_tool class="PyPackageRequirementsInspection" enabled="true" level="WARNING" enabled_by_default="true">
+      <option name="ignoredPackages">
+        <value>
+          <list size="4">
+            <item index="0" class="java.lang.String" itemvalue="scipy" />
+            <item index="1" class="java.lang.String" itemvalue="joblib" />
+            <item index="2" class="java.lang.String" itemvalue="numpy" />
+            <item index="3" class="java.lang.String" itemvalue="Cython" />
+          </list>
+        </value>
+      </option>
+    </inspection_tool>
+  </profile>
+</component>
\ No newline at end of file
diff --git a/.idea/inspectionProfiles/profiles_settings.xml b/.idea/inspectionProfiles/profiles_settings.xml
new file mode 100644
index 00000000..105ce2da
--- /dev/null
+++ b/.idea/inspectionProfiles/profiles_settings.xml
@@ -0,0 +1,6 @@
+<component name="InspectionProjectProfileManager">
+  <settings>
+    <option name="USE_PROJECT_PROFILE" value="false" />
+    <version value="1.0" />
+  </settings>
+</component>
\ No newline at end of file
diff --git a/.idea/misc.xml b/.idea/misc.xml
new file mode 100644
index 00000000..3e210f59
--- /dev/null
+++ b/.idea/misc.xml
@@ -0,0 +1,7 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="Black">
+    <option name="sdkName" value="Python 3.8 (stockPredict)" />
+  </component>
+  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.8 (stockPredict)" project-jdk-type="Python SDK" />
+</project>
\ No newline at end of file
diff --git a/.idea/stockPredict.iml b/.idea/stockPredict.iml
new file mode 100644
index 00000000..aad402c4
--- /dev/null
+++ b/.idea/stockPredict.iml
@@ -0,0 +1,10 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module version="4">
+  <component name="PyDocumentationSettings">
+    <option name="format" value="PLAIN" />
+    <option name="myDocStringFormat" value="Plain" />
+  </component>
+  <component name="TestRunnerService">
+    <option name="PROJECT_TEST_RUNNER" value="py.test" />
+  </component>
+</module>
\ No newline at end of file
diff --git a/.idea/vcs.xml b/.idea/vcs.xml
new file mode 100644
index 00000000..228f533a
--- /dev/null
+++ b/.idea/vcs.xml
@@ -0,0 +1,7 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="$PROJECT_DIR$" vcs="Git" />
+    <mapping directory="$PROJECT_DIR$/MachineLearningStocks" vcs="Git" />
+  </component>
+</project>
\ No newline at end of file
diff --git a/stock_prediction.py b/stock_prediction.py
index 14a0e689..76daf358 100644
--- a/stock_prediction.py
+++ b/stock_prediction.py
@@ -4,10 +4,23 @@
 
 
 # The percentage by which a stock has to beat the S&P500 to be considered a 'buy'
-OUTPERFORMANCE = 10
+#OUTPERFORMANCE = 90
 
+def get_outer_performance():
+    while True:
+        try:
+            x = int(input("Enter min outer performance: "))
+        except ValueError:
+            print("Not an integer!")
+            continue
+        else:
+            if (x < 0 or x >100):
+                print("Valuse must be >=0 and <=100")
+                continue
+            else:
+                return (x)
 
-def build_data_set():
+def build_data_set(outer_performance):
     """
     Reads the keystats.csv file and prepares it for scikit-learn
     :return: X_train and y_train numpy arrays
@@ -22,7 +35,7 @@ def build_data_set():
         status_calc(
             training_data["stock_p_change"],
             training_data["SP500_p_change"],
-            OUTPERFORMANCE,
+            outer_performance,
         )
     )
 
@@ -30,7 +43,9 @@ def build_data_set():
 
 
 def predict_stocks():
-    X_train, y_train = build_data_set()
+    outer_performance = get_outer_performance()
+
+    X_train, y_train = build_data_set(outer_performance)
     # Remove the random_state parameter to generate actual predictions
     clf = RandomForestClassifier(n_estimators=100, random_state=0)
     clf.fit(X_train, y_train)
@@ -49,7 +64,7 @@ def predict_stocks():
     else:
         invest_list = z[y_pred].tolist()
         print(
-            f"{len(invest_list)} stocks predicted to outperform the S&P500 by more than {OUTPERFORMANCE}%:"
+            f"{len(invest_list)} stocks predicted to outperform the S&P500 by more than {outer_performance}%:"
         )
         print(" ".join(invest_list))
         return invest_list