diff --git a/main.py b/main.py
index 8681d78ee9bc4082e1bbe6063f68d66b12e0b114..1d8eca0240720512b84e2821673c595619a300c7 100644
--- a/main.py
+++ b/main.py
@@ -39,7 +39,7 @@ def cahn_hilliard_spectral(it, c):
     # wavenumbers
     kr = power(2 * pi * fft.rfftfreq(nn, d=dx), 2)
     kc = power(2 * pi * fft.fftfreq(nn, d=dx), 2)
-    kk = array([kr,]*nn) + array([kc,]*int((nn/2)+1)) .transpose()
+    kk = array([kr,]*nn) + array([kc,]*int((nn/2)+1)).transpose()
 
     for _ in range(it): 
         
@@ -61,8 +61,8 @@ if __name__ == "__main__":
     
     fig, (ax1, ax2) = plt.subplots(1, 2)
     fig.suptitle(f"Cahn-Hilliard integration for {it} iterations ")
-    ax1.contourf(xy, xy, c1, cmap="jet")
+    ax1.imshow(c1, cmap="jet", vmin=-1, vmax=1)
     ax1.set_title("Finite differences method")
-    ax2.contourf(xy, xy, c2 , cmap="jet")
+    ax2.imshow(c2, cmap="jet", vmin=-1, vmax=1)
     ax2.set_title("Spectral method")
     plt.show()