diff --git a/OCAT/fast_similarity_matching.py b/OCAT/fast_similarity_matching.py
index e0f36c0..9d4919d 100644
--- a/OCAT/fast_similarity_matching.py
+++ b/OCAT/fast_similarity_matching.py
@@ -7,7 +7,6 @@
 ##############################
 # Imports
 import numpy as np
-#from numba import njit, jit
 #np.random.seed(42)
 ##############################
 
diff --git a/OCAT/utils.py b/OCAT/utils.py
index 7fab333..045bc56 100644
--- a/OCAT/utils.py
+++ b/OCAT/utils.py
@@ -1,8 +1,7 @@
 import numpy as np
 import os
-from fast_similarity_matching import FSM
+from .fast_similarity_matching import FSM
 import faiss
-#import h5py
 from scipy import *
 from scipy.sparse import *
 import scipy.sparse
diff --git a/dist/OCAT-0.1.1.tar.gz b/dist/OCAT-0.1.1.tar.gz
index 7c23b0b..ea595a5 100644
Binary files a/dist/OCAT-0.1.1.tar.gz and b/dist/OCAT-0.1.1.tar.gz differ
diff --git a/dist/OCAT-0.1.tar.gz b/dist/OCAT-0.1.tar.gz
deleted file mode 100644
index c0c1fae..0000000
Binary files a/dist/OCAT-0.1.tar.gz and /dev/null differ
diff --git a/setup.py b/setup.py
index d0a74a2..4c86051 100644
--- a/setup.py
+++ b/setup.py
@@ -3,13 +3,13 @@
 setup(
   name = 'OCAT',
   packages = ['OCAT'],
-  version = '0.1.1',
+  version = '0.1.2',
   license='MIT',
   description = 'A new single-cell analytics framework',
   author = '',
   author_email = '',
   url = 'https://github.com/bowang-lab/OCAT',   # Provide either the link to your github or to your website
-  download_url = 'https://github.com/bowang-lab/OCAT/archive/v_01.tar.gz',    # I explain this later on
+  download_url = 'https://github.com/bowang-lab/OCAT/archive/v_0.1.2.tar.gz',    # I explain this later on
   keywords = ['RNA-SEQ', 'CLUSTERING', 'INTEGRATION'],
   install_requires=[
           'numpy',