diff --git a/mspca.py b/mspca.py
index 961fb0d..487a6d9 100644
--- a/mspca.py
+++ b/mspca.py
@@ -127,7 +127,7 @@ def predict(self, test_x, scale=True):
         if scale:
             x_pred = self.scaler.inverse_transform(res)
         else:
-            x_pred = result
+            x_pred = res
 
         return x_pred
 
@@ -161,7 +161,7 @@ def fit_transform(self, x, wavelet_func='db4', threshold=0.3, scale=True):
         if scale:
             x_hat = self.scaler.inverse_transform(res)
         else:
-            x_hat = result
+            x_hat = res
 
         self.fit_bool = True
 
diff --git a/setup.py b/setup.py
new file mode 100644
index 0000000..e5d2906
--- /dev/null
+++ b/setup.py
@@ -0,0 +1,21 @@
+from setuptools import setup, find_packages
+
+setup(
+    name='mspca',
+    version='0.0.4',
+    py_modules=['mspca'],
+    install_requires=[
+        'PyWavelets>=1.1.1',
+        'numpy',
+        'pandas',
+        'scikit-learn',
+    ],
+    
+    author='',
+    author_email='',
+    description='mspca',
+    long_description=open('README.md').read(),
+    long_description_content_type='text/markdown',
+    url='https://github.com/ptrajdos/mspca',
+    python_requires='>=3.6',
+)