diff --git a/README.md b/README.md
index 0133d8f69ae5c4c66b0c4b4cbd7ecc427936ae76..373780304cb52013b1198cb8a9e549ea405b6f98 100644
--- a/README.md
+++ b/README.md
@@ -5,6 +5,8 @@ pyTurbulence is a Python package for generating synthetic compressible turbulent
 TODO:
 - Add dilatational velocity fluctuations
 
+![alt text](image.png) ![alt text](image-1.png)
+
 ## Features
 
 - Generate solenoidal velocity fields using the method of Rogallo.
diff --git a/pyTurbulence/solver.py b/pyTurbulence/solver.py
index 53beea3c1d3f8889a98d808185f27476dbb57b17..a5f3201ec3ec6e8c38d1d4c3dca1d09ca8b1db9c 100644
--- a/pyTurbulence/solver.py
+++ b/pyTurbulence/solver.py
@@ -1,6 +1,7 @@
 import numpy as np
 from pyTurbulence.syntheticTurbulence import compute_solenoidal_velocities, compute_solenoidal_pressure, compute_thermodynamic_fields
 from pyTurbulence.spectrum import compute_tke_spectrum, plot_spectrum, energy_spectrum
+from pyTurbulence.plot3D import plot3D
 import time
 import os
 
@@ -44,11 +45,15 @@ def solve(params, results_dir):
     start = time.time()
     knyquist, wave_numbers, tke_spectrum = compute_tke_spectrum(u, v, w, domain_size[0], domain_size[1], domain_size[2])
     real_spectrum = energy_spectrum(spectrum, wave_numbers, urms, k0)
-    spectrum_path = os.path.join(results_dir, "spectrum.png")
+    spectrum_path = os.path.join(results_dir, "spectrum.pdf")
     plot_spectrum(wmax, wave_numbers, tke_spectrum, real_spectrum=real_spectrum, save_path=spectrum_path)
     end = time.time()
     spectrum_time = end - start
 
+    # Plot the 3D fields
+    # plot3D(u, domain_size, domain_resolution, name="u")
+
+
     # Save the data
     start = time.time()
     u.tofile(os.path.join(results_dir, "u.dat"))