diff --git a/README.md b/README.md
index c8cd752..e5c77b6 100644
--- a/README.md
+++ b/README.md
@@ -5,7 +5,7 @@ python implementation of the paper: "Efficient Image Dehazing with Boundary Cons
 
 ### method 1
   ```
-  pip install image_dehazer
+  pip install git+https://github.com/XavierJiezou/Single-Image-Dehazing-Python.git
   ```
   
   **Usage:**
diff --git a/image_dehazer/__init__.py b/image_dehazer/__init__.py
index 9c54b81..969b52d 100644
--- a/image_dehazer/__init__.py
+++ b/image_dehazer/__init__.py
@@ -32,12 +32,12 @@ def __AirlightEstimation(self, HazeImg):
     def __BoundCon(self, HazeImg):
         if (len(HazeImg.shape) == 3):
 
-            t_b = np.maximum((self._A[0] - HazeImg[:, :, 0].astype(float)) / (self._A[0] - self.C0),
-                             (HazeImg[:, :, 0].astype(float) - self._A[0]) / (self.C1 - self._A[0]))
-            t_g = np.maximum((self._A[1] - HazeImg[:, :, 1].astype(float)) / (self._A[1] - self.C0),
-                             (HazeImg[:, :, 1].astype(float) - self._A[1]) / (self.C1 - self._A[1]))
-            t_r = np.maximum((self._A[2] - HazeImg[:, :, 2].astype(float)) / (self._A[2] - self.C0),
-                             (HazeImg[:, :, 2].astype(float) - self._A[2]) / (self.C1 - self._A[2]))
+            t_b = np.maximum((self._A[0] - HazeImg[:, :, 0].astype(float)) / (self._A[0] - self.C0 + 1e-6),
+                             (HazeImg[:, :, 0].astype(float) - self._A[0]) / (self.C1 - self._A[0] + 1e-6))
+            t_g = np.maximum((self._A[1] - HazeImg[:, :, 1].astype(float)) / (self._A[1] - self.C0 + 1e-6),
+                             (HazeImg[:, :, 1].astype(float) - self._A[1]) / (self.C1 - self._A[1] + 1e-6))
+            t_r = np.maximum((self._A[2] - HazeImg[:, :, 2].astype(float)) / (self._A[2] - self.C0 + 1e-6),
+                             (HazeImg[:, :, 2].astype(float) - self._A[2]) / (self.C1 - self._A[2] + 1e-6))
 
             MaxVal = np.maximum(t_b, t_g, t_r)
             self._Transmission = np.minimum(MaxVal, 1)
@@ -299,4 +299,4 @@ def remove_haze(HazeImg, airlightEstimation_windowSze=15, boundaryConstraint_win
                             regularize_lambda=regularize_lambda, sigma=sigma, delta=delta,
                             showHazeTransmissionMap=showHazeTransmissionMap)
     HazeCorrectedImg, HazeTransmissionMap = Dehazer.remove_haze(HazeImg)
-    return (HazeCorrectedImg, HazeTransmissionMap)
\ No newline at end of file
+    return (HazeCorrectedImg, HazeTransmissionMap)
diff --git a/setup.py b/setup.py
new file mode 100644
index 0000000..89f15de
--- /dev/null
+++ b/setup.py
@@ -0,0 +1,24 @@
+from setuptools import setup, find_packages
+with open("README.md", "r") as fh:
+    long_description = fh.read()
+
+setup(
+     name='image_dehazer',
+     version='0.0.9',
+     author="utkarsh-deshmukh",
+     author_email="utkarsh.deshmukh@gmail.com",
+     description="remove haze from images",
+     long_description=long_description,
+     long_description_content_type="text/markdown",
+     url="https://github.com/Utkarsh-Deshmukh/Single-Image-Dehazing-Python",
+     download_url="https://github.com/Utkarsh-Deshmukh/Single-Image-Dehazing-Python/archive/master.zip",
+     install_requires=['numpy', 'opencv-python', 'scipy'],
+     license='MIT',
+     keywords=['Single Image Dehazing', 'Haze Removal'],
+     packages=find_packages(),
+     classifiers=[
+         "Programming Language :: Python :: 3",
+         "License :: OSI Approved :: MIT License",
+         "Operating System :: OS Independent",
+     ],
+ )
\ No newline at end of file