diff --git a/membrane_curvature/surface.py b/membrane_curvature/surface.py
index b418556..3403b2d 100644
--- a/membrane_curvature/surface.py
+++ b/membrane_curvature/surface.py
@@ -167,7 +167,7 @@ def interpolation_by_array(array_surface):
 
 def surface_interpolation(array_surface):
     """
-        Calculates interpolation 
+        Calculates interpolation
 
     Parameters
     ----------
@@ -179,7 +179,7 @@ def surface_interpolation(array_surface):
     -------
 
     Returns interpolated surface
-        Interpolated surface derived from set of coordinates 
+        Interpolated surface derived from set of coordinates
         in grid of `x_range, y_range` dimensions.
         Numpy array of floats of shape (`n_x_bins`, `n_y_bins`)
 
diff --git a/membrane_curvature/tests/test_membrane_curvature.py b/membrane_curvature/tests/test_membrane_curvature.py
index e59e6c8..722dee9 100644
--- a/membrane_curvature/tests/test_membrane_curvature.py
+++ b/membrane_curvature/tests/test_membrane_curvature.py
@@ -4,7 +4,8 @@
 
 
 import pytest
-from membrane_curvature.surface import interpolation_by_array, normalized_grid, derive_surface, get_z_surface, surface_interpolation
+from membrane_curvature.surface import (interpolation_by_array, normalized_grid, 
+                                        derive_surface, get_z_surface, surface_interpolation)
 from membrane_curvature.curvature import mean_curvature, gaussian_curvature
 import numpy as np
 from numpy.testing import assert_almost_equal
@@ -222,7 +223,7 @@ def test_get_z_surface(x_bin, y_bin, x_range, y_range, expected_surface):
      np.array([[150., 150,  150.,  150.],
                [150., 150., 150.,  150.],
                [150., 150., 150.,  150.]])),
-    # array 4x4 ith all 150 and two nans
+    # array 4x4 with all 150 and two nans
     (np.array([[150., 150,  150.,  150.],
                [150., np.nan, np.nan,  150.],
                [150., 130., 140.,  150.],
@@ -231,7 +232,7 @@ def test_get_z_surface(x_bin, y_bin, x_range, y_range, expected_surface):
                [150., 140., 145.,  150.],
                [150., 130., 140.,  150.],
                [150., 150., 150.,  150.]])),
-    # array 3x3  with lots of nans)
+    # array 3x3  with lots of nans
     (np.array([[np.nan, np.nan, 150.],
               [150, np.nan, 150.],
               [np.nan, 150., np.nan]]),