diff --git a/definition.py b/definition.py
index e133b08fae6a7f90728af3be0cf6acf646fa0479..4958740d3917f29e60675e4542d7cdf9d215cb17 100644
--- a/definition.py
+++ b/definition.py
@@ -19,14 +19,6 @@ class Plate:
         self.theta = theta
         self.orientation = orientation
 
-def coef(real, imag):
-    a = 0
-    if real != 0:
-        a = real
-    if imag != 0:
-        a+= imag*1j
-    return(a)
-
 class Photon:
     def __init__(self, state0, state1):
         self.state0 = [state0]
@@ -36,24 +28,24 @@ class Photon:
         print(self.state0[-1], "|0> + ", self.state1[-1], "|1>")
 
     def round_state(self,decimal,i):
-        s_0r = round((self.state0[i]).real,decimal) 
+        s_0r = round((self.state0[i]).real,decimal)
         s_0i = round((self.state0[i]).imag,decimal)
         s_1r = round((self.state1[i]).real,decimal)
-        s_1i = round((self.state1[i]).imag,decimal) 
+        s_1i = round((self.state1[i]).imag,decimal)
         return(s_0r, s_0i, s_1r, s_1i)
 
     def is_2D(self,i):
-        s_0r = round((self.state0[i]).real,3) 
+        s_0r = round((self.state0[i]).real,3)
         s_0i = round((self.state0[i]).imag,3)
         s_1r = round((self.state1[i]).real,3)
-        s_1i = round((self.state1[i]).imag,3) 
+        s_1i = round((self.state1[i]).imag,3)
         return (s_0r*s_0i == 0 and s_1r*s_1i == 0 and s_0r*s_1i == 0 and s_1r*s_0i == 0)
 
     def pur(self,i):
         rho0, phi0 = polar(self.state0[i])
         rho1, phi1 = polar(self.state1[i])
         return (2*acos(rho0), phi1-phi0)
-    
+
     def gate(self,plate):
         alpha = self.state0[-1]
         beta = self.state1[-1]
@@ -72,7 +64,7 @@ class Photon:
                 self.state1.append(1/2*(beta + 1j*alpha))
 
     def representation(self,i):
-        
+
         #Bloch Sphere
 
         fig = plt.figure()
diff --git a/interface.py b/interface.py
index 2ace76072b6791c5b6ada2e6b80ace301d6963ea..aaeb2d1a1e5b0a7f1325bf891bcf4e931e103dfd 100644
--- a/interface.py
+++ b/interface.py
@@ -155,7 +155,7 @@ def optical_path(window, photon):
     length = len(photon.state0)
     for i in range(length):
         time = Time(i,photon,window)
-        time.do()       
+        time.do()
         if i != length-1:
             plate_number = Label(window, text = 'Plate' + str(i+1), font = ('Bahnschrift Semibold','15'), fg = '#000000', bg = '#ffffff', relief = 'raised')
             plate_number.grid(column = 2*i+1, row = 0)
@@ -206,4 +206,4 @@ def graphical_grid_init():
     button_simulate.grid(column = 7, row = 6)
     root.mainloop()
 
-graphical_grid_init()
\ No newline at end of file
+graphical_grid_init()