diff --git a/CADETMatch.egg-info/SOURCES.txt b/CADETMatch.egg-info/SOURCES.txt
index 86adbbc..fc4f3a2 100644
--- a/CADETMatch.egg-info/SOURCES.txt
+++ b/CADETMatch.egg-info/SOURCES.txt
@@ -1,7 +1,6 @@
 README.md
 setup.cfg
 setup.py
-CADETMatch/__init__.py
 CADETMatch/__main__.py
 CADETMatch/autocorr.py
 CADETMatch/cache.py
diff --git a/CADETMatch.egg-info/requires.txt b/CADETMatch.egg-info/requires.txt
index 308ed1e..788cd38 100644
--- a/CADETMatch.egg-info/requires.txt
+++ b/CADETMatch.egg-info/requires.txt
@@ -14,3 +14,4 @@ pandas
 h5py
 CADET
 seaborn
+sklearn
diff --git a/CADETMatch.egg-info/top_level.txt b/CADETMatch.egg-info/top_level.txt
index 83fb8d3..8b13789 100644
--- a/CADETMatch.egg-info/top_level.txt
+++ b/CADETMatch.egg-info/top_level.txt
@@ -1 +1 @@
-CADETMatch
+
diff --git a/CADETMatch/CADETMatch.pyproj b/CADETMatch/CADETMatch.pyproj
index 33f9a18..a56a6be 100644
--- a/CADETMatch/CADETMatch.pyproj
+++ b/CADETMatch/CADETMatch.pyproj
@@ -7,14 +7,14 @@
     <StartupFile>match.py</StartupFile>
     <SearchPath>
     </SearchPath>
-    <WorkingDirectory>.</WorkingDirectory>
+    <WorkingDirectory>..</WorkingDirectory>
     <OutputPath>.</OutputPath>
     <Name>CADETMatch</Name>
     <RootNamespace>CADETMatch</RootNamespace>
     <IsWindowsApplication>False</IsWindowsApplication>
-    <InterpreterId>Global|ContinuumAnalytics|Anaconda37-64</InterpreterId>
+    <InterpreterId>CondaEnv|CondaEnv|CADETMatch</InterpreterId>
     <LaunchProvider>Standard Python launcher</LaunchProvider>
-    <CommandLineArguments>"F:\temp\example\fit\example.json"  2</CommandLineArguments>
+    <CommandLineArguments>"F:\DS9\fractionation\NSGA3_binding_fractions.json"  2</CommandLineArguments>
     <EnableNativeCodeDebugging>False</EnableNativeCodeDebugging>
     <InterpreterArguments>
     </InterpreterArguments>
@@ -390,7 +390,7 @@
     <Compile Include="video_spearman.py">
       <SubType>Code</SubType>
     </Compile>
-    <Compile Include="__init__.py">
+    <Compile Include="init__.py">
       <SubType>Code</SubType>
     </Compile>
   </ItemGroup>
@@ -405,6 +405,7 @@
     <Folder Include="search\retired\" />
   </ItemGroup>
   <ItemGroup>
+    <InterpreterReference Include="CondaEnv|CondaEnv|CADETMatch" />
     <InterpreterReference Include="Global|ContinuumAnalytics|Anaconda37-64" />
     <InterpreterReference Include="Global|PythonCore|3.7" />
   </ItemGroup>
diff --git a/setup.py b/setup.py
index de50ea5..b37ffa3 100644
--- a/setup.py
+++ b/setup.py
@@ -29,7 +29,8 @@
           'pandas',
           'h5py',
           'CADET',
-          'seaborn'
+          'seaborn',
+          'sklearn'
       ],
     classifiers=[
         "Programming Language :: Python :: 3",