diff --git a/example/sim_demo1.py b/example/sim_demo1.py
index dee5c4f89969448a0029c0a692db31834e3b1795..28831f736ff37922e099c684dddafd3cf2b90db9 100644
--- a/example/sim_demo1.py
+++ b/example/sim_demo1.py
@@ -2,6 +2,9 @@
 from SpecGen.SpecGenerator import SpecGenerator
 from SpecGen.Config import Config
 
+from astropy.io import fits
+from astropy.table import Table
+
 import galsim
 
 
@@ -15,4 +18,8 @@ if __name__ == '__main__':
     specG = SpecGenerator(sedFn = sedFn, grating = 'GI', beam = 'A', aper = 2.0, xcenter = 5000,ycenter = 5000, p_size = 0.074, psf = psf, skybg = 0.3, dark = 0.02, readout = 5, t = 150, expNum = 1,config = config)
     specTab, specImg, img, fluxRa=specG.generateSpec1dforStar()
 
+    fits.writeto("specImg.fits",specImg,overwrite=True)
+    fits.writeto("DImg.fits",img,overwrite=True)
+    specTab.write("specTab.fits",overwrite=True)
+